hdu5820 询问简单路径 主席树

题意：在一个大小为50000*50000的矩形中，有n个路灯。（n<=500000）

询问是否每一对路灯之间存在一条道路，使得长度为|x1 – x2| + |y1 – y2|且每个拐弯点都是路灯。

ps注意 实现的时候下方y要+1(边界问题) 主席树root维是记录的x棵树的前缀的树根，每棵树记录的是y的分布情况

#include <bits/stdc++.h>
using namespace std;
#define maxn 9000000
#define lson l,mid,rt<<1
#define rson mid,r+1,rt<<1|1
#define getmid int mid=(l+r)>>1;
#define mem(x,v) memset(x,v,sizeof(x));
#define ft first
#define sd second
#define mp make_pair
#define pii pair<int,int>
int n,flag;
map<pii,int>myp;
int siz;
int sum[maxn],ls[maxn],rs[maxn];
int root[51000];
void build(int &rt,int l,int r)
{
    siz++;rt=siz;
    sum[rt]=0;
    if (l==r) return;
    getmid
    build(ls[rt],l,mid);
    build(rs[rt],mid+1,r);
    sum[rt]=sum[ls[rt]]+sum[rs[rt]];
}
void update(int l,int r,int x,int &y,int pos,int val)
{
    siz++;y=siz;
    if (l==r)
    {
        sum[y]=sum[x]+val;
        return;
    }
    ls[y]=ls[x]; rs[y]=rs[x];
    getmid
    if (pos <= mid) update(l,mid,ls[x],ls[y],pos,val);
    else update(mid+1,r,rs[x],rs[y],pos,val);
    sum[y]=sum[ls[y]]+sum[rs[y]];
}

int query(int l,int r,int rt,int x,int y)
{
    if (x<=l && r<=y)
    {
        return sum[rt];
    }
    getmid
    int res=0;
    if (x <= mid) res+=query(l,mid,ls[rt],x,y);
    if (mid <  y) res+=query(mid+1,r,rs[rt],x,y);
    return res;
}
int mx[51000],my[51000];
pii po[501000];
void solve()
{
    mem(mx,0);mem(my,0);
    siz=0;
    build(root[0],1,50000);
    int last=root[0],x1,x2,y1,y2,sum1=0,sum2=0;
    for (int i=1;i<=n;i++)
    {
        x1=po[i].ft; y1=po[i].sd;
        x2=po[my[y1]].ft; y2=po[mx[x1]].sd;

        update(1,50000,last,root[x1],y1,1);
        sum1=query(1,50000,root[x1],y2+1,y1);
        sum2=query(1,50000,root[x2],y2+1,y1);
        if(sum1-sum2!=1)
        {
            flag=0;
            return;
        }
        mx[x1]=i; my[y1]=i; last=root[x1];
    }
}

int main()
{
    int T;
    while (~scanf("%d",&n))
    {
        if (n==0) break;
        myp.clear();
        int num=0;
        for (int i=1;i<=n;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            if(!myp[mp(x,y)])
            {
                myp[mp(x,y)]=1;
                num++;
                po[num]=mp(x,y);
            }
        }
        n=num;flag=1;
        sort(po+1,po+n+1);

        solve();

        for (int i=1;i<=n;i++) po[i].ft=50001-po[i].ft;
        sort(po+1,po+n+1);

        solve();
        printf("%s\n",flag ? "YES" : "NO" );
    }
}
/*
5
1 1
1 2
1 3
2 1
3 3
*/