104332024-04-02 10:57:45szilÚthálózatbővítéscpp17Runtime error 70/100435ms63688 KiB
#include <bits/stdc++.h>

using namespace std;
using ll = long long;

struct Point {
    ll x, y;

    bool operator==(const Point &p) {
        return x == p.x && y == p.y;
    }
};

struct Line {
    int u, v;
};

ll cross(Point p, Point q) {
    return p.x * q.y - p.y * q.x;
}

ll squared(Point p) {
    return p.x*p.x+p.y*p.y;
}

ll forog(Point a, Point b, Point c) {
    b.x -= a.x; c.x -= a.x;
    b.y -= a.y; c.y -= a.y;
    ll o = cross(b, c);
    if (o > 0) return 1;
    if (o == 0) return 0;
    if (o < 0) return -1;
}

bool operator<(const Point &p, const Point &q) {
    if ((p.x >= 0) == (q.x >= 0)) {
        ll o = cross(p, q);
        if (o == 0) return squared(p) < squared(q);
        return cross(p, q) > 0;
    }
    return (p.x >= 0) < (q.x >= 0);
}

const int MAXN = 200'001;

Point points[MAXN];
Line lines[MAXN];
bool ans[MAXN];
vector<Line> to_remove[MAXN], to_add[MAXN];
set<Line> s;
int n, ox, oy;

bool operator<(const Line &a, const Line &b) {
    Point p1 = points[a.u], q1 = points[a.v], p2 = points[b.u], q2 = points[b.v];
    if (cross(p1, q1) < 0) swap(p1, q1);
    if (cross(p2, q2) < 0) swap(p2, q2);
    if (forog(p1, q1, p2) == 0) {
        return forog(p1, q1, q2) < 0;
    }
    if (forog(p1, q1, q2) == 0) {
        return forog(p1, q1, p2) < 0;
    }
    if (forog(p1, q1, p2) == forog(p1, q1, q2)) return forog(p1, q1, p2) < 0;
    if (forog(p2, q2, p1) == forog(p2, q2, q1)) return forog(p2, q2, p1) > 0;
    return false;
}

void solve() {
    vector<int> indices(n);
    iota(indices.begin(), indices.end(), 1);
    sort(indices.begin(), indices.end(), [&](int i, int j){
        return points[i] < points[j];
    });
    for (int i = 0; i < n; ) {
        int j = i;
        while (j + 1 < n && cross(points[indices[i]], points[indices[j + 1]]) == 0) j++;
        for (int k = i; k <= j; k++) {
            for (Line l : to_remove[indices[k]]) {
                s.erase(l);
            }
        }
        if (s.lower_bound({indices[i], indices[i]}) == s.begin()) {
            ans[indices[i]] = true;
        }
        for (int k = i; k <= j; k++) {
            for (Line l : to_add[indices[k]]) {
                assert(s.insert(l).second);
            }
        }
        i = j + 1;
    }    
}

int main() {
    ios::sync_with_stdio(0); cin.tie(0);
    cin >> n >> ox >> oy;
    for (int i = 1; i <= n; i++) {
        cin >> points[i].x >> points[i].y;
        points[i].x -= ox;
        points[i].y -= oy;
    }
    for (int i = 0; i < n-1; i++) {
        int &u = lines[i].u, &v = lines[i].v;
        cin >> u >> v;
        if (points[v] < points[u]) swap(u, v);

        ll o = cross(points[u], points[v]);
        if (o > 0) {
            to_add[u].emplace_back(lines[i]);
            to_remove[v].emplace_back(lines[i]);
        } else if (o < 0) {
            s.insert(lines[i]);
            to_remove[u].emplace_back(lines[i]);
            to_add[v].emplace_back(lines[i]);
        }
    }
    solve();
    cout << count(ans+1, ans+n+1, true) << "\n";
    for (int i = 1; i <= n; i++) {
        if (ans[i]) cout << i << " ";
    }
    return 0;
}
SubtaskSumTestVerdictTimeMemory
subtask10/0
1Accepted9ms20688 KiB
2Accepted133ms33100 KiB
subtask210/10
3Accepted10ms22884 KiB
4Accepted9ms23168 KiB
5Accepted9ms23396 KiB
6Accepted12ms23628 KiB
7Accepted10ms23968 KiB
subtask320/20
8Accepted8ms23720 KiB
9Accepted9ms23736 KiB
10Accepted10ms24080 KiB
11Accepted17ms25152 KiB
12Accepted28ms26056 KiB
13Accepted115ms33444 KiB
14Accepted236ms43988 KiB
15Accepted18ms29600 KiB
16Accepted41ms34836 KiB
17Accepted100ms39404 KiB
subtask410/10
18Accepted12ms30612 KiB
19Accepted17ms31444 KiB
20Accepted14ms31304 KiB
21Accepted20ms32840 KiB
22Accepted20ms32356 KiB
subtask530/30
23Accepted13ms31824 KiB
24Accepted28ms33688 KiB
25Accepted12ms32048 KiB
26Accepted12ms32192 KiB
27Accepted19ms33392 KiB
28Accepted32ms34732 KiB
29Accepted46ms36216 KiB
30Accepted26ms35336 KiB
31Accepted43ms39896 KiB
32Accepted115ms43312 KiB
subtask60/30
33Accepted64ms45340 KiB
34Accepted35ms39916 KiB
35Accepted228ms52760 KiB
36Accepted259ms53028 KiB
37Accepted273ms53028 KiB
38Accepted347ms55928 KiB
39Accepted372ms56720 KiB
40Accepted435ms58304 KiB
41Runtime error246ms63688 KiB
42Accepted239ms39176 KiB