BZOJ 1018 堵塞的交通traffic(线段树)

题目链接：http://61.187.179.132/JudgeOnline/problem.php?id=1018

题意：一个2*n的格子，相邻格子之间有一条道路。初始时道路是不通的。 三种操作：（1）某条道路连起来；（2）某条道路断开；（3）询问某两个格子是否相连。

思路：（1）首先，线段树的节点保存[L,R]区间两端的四个格子的连通情况，共六种,另外还要保存[L,mid],[mid+1,R]这两段之间的连通情况，即格子(1,mid)和(1,mid+1)以及(2,mid)和(2,mid+1)是否连接；

（2）查询[L,R]时，有可能L直接与R相连，也有可能L向前走再回来，或者R向后走再回来。查询p=[1,L],r=[L,R],q=[R,n]。那么p的右侧两个格子相连或者r的左侧两个格子相连，则r的左侧两个格子相连；q的左侧两个格子相连或者r的右侧两个格子相连，则r的右侧两个格子相连。此时利用r的四个格子的连通情况就可以判断询问的两个格子是否连通。

（3）修改时，x1=x2，则两个格子为同一行相邻格子，否则为同一列的两个格子。

struct Node
{
    int lb,rb,b[2][2];
};

struct node
{
    int b0,b1,L,R,mid;
    Node b;
};

node a[N<<2];

Node merge(Node a,Node b,int b0,int b1)
{
    Node ans;
    ans.lb=a.lb||(a.b[0][0]&&b0&&b.lb&&b1&&a.b[1][1]);
    ans.rb=b.rb||(b.b[0][0]&&b0&&a.rb&&b1&&b.b[1][1]);
    clr(ans.b,0);
    int i,j;
    if(b0) FOR0(i,2) FOR0(j,2) ans.b[i][j]|=a.b[i][0]&&b.b[0][j];
    if(b1) FOR0(i,2) FOR0(j,2) ans.b[i][j]|=a.b[i][1]&&b.b[1][j];
    return ans;
}

void build(int t,int L,int R)
{
    a[t].L=L;
    a[t].R=R;
    a[t].mid=(L+R)>>1;
    if(L==R)
    {
        a[t].b.b[0][0]=a[t].b.b[1][1]=1;
        return;
    }
    build(t*2,L,a[t].mid);
    build(t*2+1,a[t].mid+1,R);
}

void update(int t,int x1,int y1,int x2,int y2,int b)
{
    if(x1==x2&&a[t].mid==y1)
    {
        if(x1==0) a[t].b0=b;
        else a[t].b1=b;
        a[t].b=merge(a[t*2].b,a[t*2+1].b,a[t].b0,a[t].b1);
        return;
    }
    if(a[t].L==a[t].R)
    {
        a[t].b.lb=a[t].b.rb=a[t].b.b[0][1]=a[t].b.b[1][0]=b;
        return;
    }
    if(y1<=a[t].mid) update(t*2,x1,y1,x2,y2,b);
    else update(t*2+1,x1,y1,x2,y2,b);
    a[t].b=merge(a[t*2].b,a[t*2+1].b,a[t].b0,a[t].b1);
}

Node query(int t,int L,int R)
{
    if(a[t].L==L&&a[t].R==R) return a[t].b;
    if(R<=a[t].mid) return query(t*2,L,R);
    if(L>a[t].mid) return query(t*2+1,L,R);
    Node p=query(t*2,L,a[t].mid);
    Node q=query(t*2+1,a[t].mid+1,R);
    return merge(p,q,a[t].b0,a[t].b1);
}

int n;

int main()
{
    RD(n); build(1,1,n);
    char op[10];
    int x1,y1,x2,y2;
    Node p,q,r;
    int i,j,ans;
    while(scanf("%s",op),op[0]!='E')
    {
        RD(x1,y1); RD(x2,y2); x1--; x2--;
        if(y1>y2) swap(y1,y2),swap(x1,x2);
        if(op[0]=='C') update(1,x1,y1,x2,y2,0);
        else if(op[0]=='O') update(1,x1,y1,x2,y2,1);
        else
        {
            p=query(1,1,y1);
            q=query(1,y2,n);
            r=query(1,y1,y2);
            if(p.rb) r.lb=1;
            if(q.lb) r.rb=1;
            ans=0;
            FOR0(i,2) FOR0(j,2) if(r.b[i][j])
            {
                if((i==x1||r.lb)&&(j==x2||r.rb)) ans=1;
            }
            if(ans) puts("Y");
            else puts("N");
        }
    }
}