【BZOJ4049】【Cerc2014】 Mountainous landscape 【凸包】【线段树】

是时候补补计算几何了。。
bzoj链接

题目大意

给你一条折线（顶点为$p_1,p_2,\dots,p_n$），求对于折线中每条线段向右延长与折线的第一个交点所在的线段（不能交于该线段的上端点）的编号（无解输出0）。

题解

用线段树维护当前区间$l$至$r$的点的上凸壳。查询时在区间$i+1,n-1$寻找线段$p_i,p_{i+1}$与那个编号最小的上凸壳相交（因为第一个交点必定在上凸壳上）。$pushup$时直接凸包暴力合并，可以用二分加向量叉积判断是否在当前区间有交点。时间复杂度两个$log$。

代码

//#pragma GCC optimize(3)
#include<iostream>
#include<iomanip>
#include<stack>
#include<queue>
#include<list>
#include<vector>
#include<map>
#include<set>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#define ull unsigned long long
#define db double
#define ll long long
#define inf 100009
#define rd(n) {n=0;char ch;int f=0;do{ch=getchar();if(ch=='-'){f=1;}}while(ch<'0'||ch>'9');while('0'<=ch&&ch<='9'){n=(n<<1)+(n<<3)+ch-48;ch=getchar();}if(f)n=-n;}
using namespace std;
int n;

struct point{
    int x,y;
    point(){}
    point(int xx,int yy){
        x=xx,y=yy;
    }
}p[inf];

struct Vector{
    int x,y;
    Vector(){}
    Vector(point xx,point yy){
        x=xx.x-yy.x;
        y=xx.y-yy.y;
    }
};

ll cal(Vector x,Vector y){
    return (ll)x.x*y.y-(ll)x.y*y.x;
}

struct ConvexHull{
    vector <point> s;
}ch[inf*4];

void CH_clear(ConvexHull &x){
    x.s.clear();
    return;
}

void CH_insert(ConvexHull &x,point p){
    int cnt=x.s.size();
    while (cnt>1 && cal(Vector(x.s[cnt-1],x.s[cnt-2]),Vector(p,x.s[cnt-2]))>=0){
        x.s.pop_back();
        cnt--;
    }
    x.s.push_back(p);
    return;
}

void CH_merge(ConvexHull &x,ConvexHull y){
    for (int i=0;i<y.s.size();i++){
        CH_insert(x,y.s[i]);
    }
    return;
}

bool CH_find(ConvexHull x,point p1,point p2){
    Vector vec=Vector(p2,p1);
    int l=0,r=x.s.size()-1;
    if (l>r){
        return 0;
    }
    while (l<r){
        int mid=(l+r)/2;
        if (cal(Vector(x.s[mid],p1),vec)<cal(Vector(x.s[mid+1],p1),vec)){
            r=mid;
        }
        else{
            l=mid+1;
        }
    }
    if (cal(Vector(x.s[l],p1),vec)<0){
        return 1;
    }
    else if (l<x.s.size()-1 && cal(Vector(x.s[l],p1),vec)<0){
        return 1;
    }
    return 0;
}

void ST_build(int u,int l,int r){
    if (l==r){
        CH_clear(ch[u]);
        CH_insert(ch[u],p[l]);
        CH_insert(ch[u],p[l+1]);
        return;
    }
    int mid=(l+r)/2;
    ST_build(u*2,l,mid);
    ST_build(u*2+1,mid+1,r);
    ch[u]=ch[u*2];
    CH_merge(ch[u],ch[u*2+1]);
    return;
}

int ST_query(int u,int l,int r,int L,int R,point p1,point p2){
    if (L<=l && r<=R){
        if (!CH_find(ch[u],p1,p2)){
            return 0;
        }
        else if (l==r){
            return l;
        }
    }
    int mid=(l+r)/2,ans=0;
    if (L<=mid){
        ans=ST_query(u*2,l,mid,L,R,p1,p2);
    }
    if (ans){
        return ans;
    }
    if (R>mid){
        return ST_query(u*2+1,mid+1,r,L,R,p1,p2);
    }
    return 0;
}

int main(){
    int T;
    rd(T)
    for (int t=1;t<=T;t++){
        rd(n)
        for (int i=1;i<=n;i++){
            rd(p[i].x) rd(p[i].y)
        }
        ST_build(1,1,n-1);
        for (int i=1;i<n;i++){
            printf("%d ",ST_query(1,1,n-1,i+1,n-1,p[i],p[i+1]));
        }
        puts("");
    }
    return 0;
}