【NOI2014模拟7.14】Problem A

Description

我们有一个树，大小为n。考虑树上的一条路径，如果一个边的两个点都在这路径上，我们称这个边属于这个路径，如果一个边有且只有一个点在这路径上，我们称这个边与这个路径相邻。现在每个边要么是黑色的要么是白色的，一开始所有边都是白色的。

我们有3个操作，将某路径反色，将与某路径相邻的所有边反色，求一个路径上黑边的总数。

Solution

这题肯定是链剖嘛~
取反简单，把相邻的取反也很简单，
发现，对于一条链，只需改两个，这个直接改，
处理完链内的情况，其余的，只有在切换链的时候会遇到更改，直接特判，

代码+++++++++QwQ~

复杂度：$O(n\log(n))$

Code

#include <iostream>
#include <cstdio>
#include <cstdlib>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define efo(i,q) for(int i=A[q];i;i=B[i][0])
using namespace std;
const int N=100500;
int read(int &n)
{
    char ch=' ';int q=0,w=1;
    for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());
    if(ch=='-')w=-1,ch=getchar();
    for(;ch>='0' && ch<='9';ch=getchar())q=q*10+ch-48;n=q*w;return n;
}
int m,n,ans;
struct qqww
{
    int si,ls,fa,c,zx,nx;
    bool f;
}a[N];
int B[2*N][2],A[N],B0,b0;
struct wwqq
{
    bool laf,f,la;
    int sum;
}b[N*4];
void link(int q,int w)
{
    B[++B0][0]=A[q];A[q]=B0,B[B0][1]=w;
    B[++B0][0]=A[w];A[w]=B0,B[B0][1]=q;
}
int dfs1(int q,int fa,int c)
{
    a[q].fa=fa;a[q].si=1;a[q].c=c;
    efo(i,q)if(B[i][1]!=fa)a[q].si+=dfs1(B[i][1],q,c+1);
    return a[q].si;
}
void dfs2(int q,int fa,int ls)
{
    a[q].zx=++b0;
    a[q].ls=(ls?ls:(ls=q));
    int mx=0;
    efo(i,q)if(B[i][1]!=fa)
        if(a[mx].si<a[B[i][1]].si)mx=B[i][1];
    a[q].nx=mx;
    if(mx)dfs2(mx,q,ls);
    efo(i,q)if(B[i][1]!=fa&&B[i][1]!=mx)dfs2(B[i][1],q,0);
}
void doit(int e,int l,int r)
{
    if(b[e].laf)
    {
        b[e].f=!b[e].f;
        if(l!=r)
        {
            b[e*2].laf=!b[e*2].laf;
            b[e*2+1].laf=!b[e*2+1].laf;
        }
        b[e].laf=0;
    }
    if(b[e].la)
    {
        b[e].sum=r-l+1-b[e].sum;
        if(l!=r)
        {
            b[e*2].la=!b[e*2].la;
            b[e*2+1].la=!b[e*2+1].la;
        }
        b[e].la=0;
    }
}
void merge(int e)
{
    b[e].sum=b[e*2].sum+b[e*2+1].sum;
}
void change(int l,int r,int e,int l1,int r1,bool l2)
{
    if((!l1)||l1>r1)return;
    int t=(l+r)>>1;
    if(l!=r)doit(e*2,l,t),doit(e*2+1,t+1,r);
    if(l==l1&&r==r1)
    {
        if(!l2)b[e].la=!b[e].la;
            else b[e].laf=!b[e].laf;
        doit(e,l,r);
        return;
    }
    if(r1<=t)change(l,t,e*2,l1,r1,l2);
        else if(t<l1)change(t+1,r,e*2+1,l1,r1,l2);
            else 
            {
                change(l,t,e*2,l1,t,l2);
                change(t+1,r,e*2+1,t+1,r1,l2);
            }
    merge(e);
}
bool findf(int l,int r,int e,int l1)
{
    if(!l1)return 0;
    if(l==r)return b[e].f;
    int t=(l+r)>>1;
    doit(e*2,l,t),doit(e*2+1,t+1,r);
    if(l1<=t)return findf(l,t,e*2,l1);
        else return findf(t+1,r,e*2+1,l1);
}
int find(int l,int r,int e,int l1,int r1)
{
    if((!l1)||l1>r1)return 0;
    int t=(l+r)>>1;
    if(l!=r)doit(e*2,l,t),doit(e*2+1,t+1,r);
    if(l==l1&&r==r1)return b[e].sum;
    if(r1<=t)return find(l,t,e*2,l1,r1);
        else if(t<l1)return find(t+1,r,e*2+1,l1,r1);
            else return find(l,t,e*2,l1,t)+find(t+1,r,e*2+1,t+1,r1);
}
void modify(int q,int w,int K)
{
    while(a[q].ls!=a[w].ls)
    {
        if(a[a[q].ls].c<a[a[w].ls].c)swap(q,w);
        int t=a[q].ls;
        if(a[t].fa&&a[t].f!=findf(1,n,1,a[a[t].fa].zx))change(1,n,1,a[t].zx,a[t].zx,0),a[t].f=!a[t].f;
        if(K)change(1,n,1,a[a[q].nx].zx,a[a[q].nx].zx,0),a[t].f=!a[t].f;
        change(1,n,1,a[t].zx,a[q].zx,K);
        q=a[t].fa;
    }
    if(a[q].c<a[w].c)swap(q,w);
    if(a[w].zx+(!K)<=a[q].zx)change(1,n,1,a[w].zx+(!K),a[q].zx,K);
    if(K)
    {
        change(1,n,1,a[w].zx,a[w].zx,0);
        change(1,n,1,a[a[q].nx].zx,a[a[q].nx].zx,0);
    }
}
int search(int q,int w)
{
    int ans=0;
    while(a[q].ls!=a[w].ls)
    {
        if(a[a[q].ls].c<a[a[w].ls].c)swap(q,w);
        int t=a[q].ls;
        if(a[t].fa&&a[t].f!=findf(1,n,1,a[a[t].fa].zx))
            change(1,n,1,a[t].zx,a[t].zx,0),a[t].f=!a[t].f;
        ans+=find(1,n,1,a[t].zx,a[q].zx);
        q=a[t].fa;
    }
    if(a[q].c<a[w].c)swap(q,w);
    ans+=find(1,n,1,a[w].zx+1,a[q].zx);
    return ans;
}
int main()
{
    int q,w,e,_;
    read(n);
    fo(i,1,n-1)read(q),read(w),link(q,w);
    dfs1(1,0,1);
    dfs2(1,0,0);
    read(_);
    while(_--)
    {
        read(e),read(q),read(w);
        if(e<3)modify(q,w,e-1);
            else printf("%d\n",search(q,w));
    }
    return 0;
}