WA on 23... Ural 1469

Posted 2020-10-17 ichneumon

http://acm.timus.ru/problem.aspx?space=1&num=1469

#include <assert.h>
#include <set>
#include <algorithm>
#include <vector>
#include <cstdio>
#include <cstring>
using namespace std;

const int N = 5e4 + 100;
const double EPS = 1e-5;
int sgn(double x) {
    return x < -EPS ? -1 : EPS < x;
}
typedef double db;


struct Poi {
    db x, y;
    Poi() {}
    Poi(db x, db y):x(x), y(y) {}
    bool operator < (const Poi&rhs) const {
        return sgn(x - rhs.x) < 0 || (sgn(x - rhs.x) == 0 && sgn(y - rhs.y) < 0);
    }
    bool operator == (const Poi&rhs) const{
        return sgn(x - rhs.x) == 0 && sgn(y - rhs.y) == 0;
    }
};

Poi operator - (Poi a, Poi b) {
    return Poi(a.x - b.x, a.y - b.y);
}


struct Seg {
    Poi a, b;
    void read() {
        scanf("%lf%lf%lf%lf", &a.x, &a.y, &b.x, &b.y);
        if (b < a) swap(a, b);
        assert(!(a == b));
    }
} s[N];

db cross(const Poi&a, const Poi&b) {
    return a.x * b.y  - a.y * b.x;
}

db dot(const Poi&a, const Poi&b) {
    return a.x * b.x + a.y * b.y;
}

bool onLeft(Seg s, Poi p) {
    return cross(p - s.a, s.b - s.a) > 0;
}

bool onSeg(Seg s, Poi p) {
    return sgn(cross(p - s.a, s.b - s.a)) == 0 && sgn(dot(p - s.a, s.b - s.a)) >= 0 && sgn(dot(p - s.b, s.a - s.b)) >= 0;
}

bool hasInter(int i, int j) {
    if (onSeg(s[i], s[j].a) || onSeg(s[i], s[j].b) || onSeg(s[j], s[i].a) || onSeg(s[j], s[i].b))
        return true;
    int a, b;
    a = onLeft(s[i], s[j].a);
    b = onLeft(s[i], s[j].b);
    if (a == b) return false;
    a = onLeft(s[j], s[i].a);
    b = onLeft(s[j], s[i].b);
    if (a == b) return false;
    return true;
}

struct Event {
    int type; // 0 for left end, 1 for right end
    int segId;
    Poi *p;
    Event() {}
    Event(int type, int segId, Poi*p):type(type), segId(segId), p(p) {}
    bool operator < (const Event&rhs) const {
        return *p < *rhs.p || (*p == *rhs.p && type < rhs.type);
    }
};
struct EventQueue:vector<Event> {
    int cur;
    Event* nextEvent() {
        if (cur < size())
            return &at(cur++);
        return NULL;
    }
} EQ;

void end(int i, int j) {
    puts("YES");
    printf("%d %d\n", i, j);
    exit(0);
}

struct TNode;

typedef set<TNode>::iterator iter;
struct TNode {
    int segId;
    TNode(int segId):segId(segId) {}
    bool operator < (const TNode&rhs) const {
        if (rhs.segId != segId && hasInter(rhs.segId, segId))
            end(segId, rhs.segId);
        int t = sgn(cross(s[rhs.segId].a - s[segId].a, s[segId].b - s[segId].a));
        return t < 0;
    }
    bool operator == (const TNode&rhs) const {
        return segId == rhs.segId;
    }
};

iter pos[N];

struct SweepLine:set<TNode> {

    bool add(Event*e, int&i, int&j) {
        pos[e->segId] = insert(TNode(e->segId)).first;
        iter it = pos[e->segId];
        iter pre = it, suf = it;
        --pre, ++suf;
        i = it->segId;
        if (pre != end()) {
            j = pre->segId;
            if (i != j && hasInter(i, j)) {
                return true;
            }
        }
        if (suf != end()) {
            j = suf->segId;
            if (i != j && hasInter(i, j)) {
                return true;
            }
        }
        return false;
    }

    bool del(Event*e, int &i, int&j) {
        iter it = pos[e->segId];
        iter pre = it, suf = it;
        --pre, ++suf;
        if (pre != end() && suf != end()) {
            i = pre->segId, j = suf->segId;
            if (i != j && hasInter(i, j))
                return true;
        }
        erase(it);
        return false;
    }

} SL;

int main() {
#ifdef lol
    freopen("1469.in", "r", stdin);
    freopen("1469.out", "w", stdout);
#endif
    int n;
    scanf("%d", &n);
    for (int i = 1; i <= n; ++i) {
        s[i].read();
    }
    
    for (int i = 1; i <= n; ++i) {
        EQ.push_back(Event(0, i, &s[i].a));
        EQ.push_back(Event(1, i, &s[i].b));
    }
    sort(EQ.begin(), EQ.end());


    Event* e;
    int i, j;
    bool ok = false;
    while (!ok && (e = EQ.nextEvent()) != NULL) {
    //    printf("%d %d\n", e->type, e->segId);
        if (e->type == 0) {
            if (SL.add(e, i, j))
                ok = true;
        } else {
            if (SL.del(e, i, j))
                ok = true;
        }
    }
    if (ok) {
        puts("YES");
        printf("%d %d\n", i, j);
    } else {
        puts("NO");
    }
    
    return 0;
}

 

写得巨丑, 有空重写一发吧.