题意:
有一棵n个点的树,一开始所有边都是白的。
m个操作:
1、将a,b路径上的边颜色取反
2、将与a,b路径相邻的边颜色取反
3、询问a,b路径上多少条边是黑的
定义e在路径上当e有且仅有一个端点在路径上
n,m<=100000
#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<iostream>
#define N 110000
using namespace std;
struct node{int y,nex;}a[N*2];
struct node1{int l,r,lc,rc,c0,c1,lz;}lt[2][2*N];
struct node2{int fa,tp,son,siz,dep;}t[N];
int n,m,id,dfn[N],las[N],fir[N],len,tl;
void ins(int x,int y)
{
a[++len].y=y;a[len].nex=fir[x];fir[x]=len;
}
void dfs1(int x,int fa)
{
t[x].fa=fa;t[x].son=0;t[x].siz=1;t[x].dep=t[fa].dep+1;
for(int k=fir[x];k;k=a[k].nex)
{
int y=a[k].y;
if(y==fa) continue;
dfs1(y,x);
t[x].siz+=t[y].siz;
if(t[t[x].son].siz<t[y].siz) t[x].son=y;
}
}
void dfs2(int x,int tp)
{
t[x].tp=tp;
dfn[x]=++id;
if(t[x].son) dfs2(t[x].son,tp);
for(int k=fir[x];k;k=a[k].nex)
{
int y=a[k].y;
if(y==t[x].fa || y==t[x].son) continue;
dfs2(y,y);
}
las[x]=id;
}
void bt(int l,int r)
{
int now=++tl;
for(int i=0;i<=1;i++)
{
lt[i][now].l=l;lt[i][now].r=r;
lt[i][now].lc=lt[i][now].rc=lt[i][now].c1=lt[i][now].lz=0;
lt[i][now].c0=r-l+1;
}
if(l<r)
{
int mid=(l+r)/2;
lt[0][now].lc=lt[1][now].lc=tl+1;bt(l,mid);
lt[0][now].rc=lt[1][now].rc=tl+1;bt(mid+1,r);
}
}
void down(int o,int now)
{
int lc=lt[o][now].lc,rc=lt[o][now].rc;
swap(lt[o][now].c0,lt[o][now].c1);
if(lc) lt[o][lc].lz^=1;
if(rc) lt[o][rc].lz^=1;
lt[o][now].lz=0;
}
void upd(int o,int now)
{
int lc=lt[o][now].lc,rc=lt[o][now].rc;
if(lt[o][lc].lz) down(o,lc);
if(lt[o][rc].lz) down(o,rc);
lt[o][now].c0=lt[o][lc].c0+lt[o][rc].c0;
lt[o][now].c1=lt[o][lc].c1+lt[o][rc].c1;
}
int find(int o,int now,int l,int r)
{
if(r==0 || l>r) return 0;
int mid=(lt[o][now].l+lt[o][now].r)/2,lc=lt[o][now].lc,rc=lt[o][now].rc;
if(lt[o][now].lz) down(o,now);
if(lt[o][now].l==l && lt[o][now].r==r) return lt[o][now].c1;
if(mid>=r) return find(o,lc,l,r);
else if(l>mid) return find(o,rc,l,r);
else return find(o,lc,l,mid)+find(o,rc,mid+1,r);
}
void change(int o,int now,int l,int r)
{
if(r==0 || l>r) return;
int mid=(lt[o][now].l+lt[o][now].r)/2,lc=lt[o][now].lc,rc=lt[o][now].rc;
if(lt[o][now].lz) down(o,now);
if(lt[o][now].l==l && lt[o][now].r==r) {lt[o][now].lz=1;return;}
if(mid>=r) change(o,lc,l,r);
else if(l>mid) change(o,rc,l,r);
else change(o,lc,l,mid),change(o,rc,mid+1,r);
upd(o,now);
}
void modify(int x,int y,int o)
{
int fx=t[x].tp,fy=t[y].tp;
while(fx!=fy)
{
if(t[fx].dep<t[fy].dep) swap(fx,fy),swap(x,y);
if(o==1) change(0,1,dfn[fx],dfn[x]);
else
{
change(1,1,dfn[fx],dfn[x]);
change(0,1,dfn[t[x].son],dfn[t[x].son]);
change(0,1,dfn[fx],dfn[fx]);
}
x=t[fx].fa;
fx=t[x].tp;
}
if(dfn[x]>dfn[y]) swap(x,y);
if(o==1)
{
if(x==y) return;
change(0,1,dfn[t[x].son],dfn[y]);
}
else
{
change(1,1,dfn[x],dfn[y]);
change(0,1,dfn[t[y].son],dfn[t[y].son]);
change(0,1,dfn[x],dfn[x]);
}
}
void query(int x,int y)
{
int fx=t[x].tp,fy=t[y].tp,ans=0;
while(fx!=fy)
{
if(t[fx].dep<t[fy].dep) swap(x,y),swap(fx,fy);
if(fx!=x) ans+=find(0,1,dfn[t[fx].son],dfn[x]);
ans+=find(0,1,dfn[fx],dfn[fx])^find(1,1,dfn[t[fx].fa],dfn[t[fx].fa]);
x=t[fx].fa;
fx=t[x].tp;
}
if(dfn[x]>dfn[y]) swap(x,y);
if(x==y)
{
printf("%d\n",ans);
return;
}
ans+=find(0,1,dfn[t[x].son],dfn[y]);
printf("%d\n",ans);
}
int main()
{
int z;scanf("%d",&z);
while(z--)
{
scanf("%d",&n);
len=0;
for(int i=1;i<=n;i++) fir[i]=0;
for(int i=1;i<n;i++)
{
int x,y;scanf("%d%d",&x,&y);
ins(x,y);ins(y,x);
}
id=0;
dfs1(1,0);
dfs2(1,1);
tl=0;
bt(1,n);
scanf("%d",&m);
while(m--)
{
int o,x,y;
scanf("%d%d%d",&o,&x,&y);
if(o==3) query(x,y);
else modify(x,y,o);
}
}
return 0;
}
题解:
突然想起这道题,觉得很妙。膜cls。
相邻似乎根本不可做,题解的做法是:
给每条边存0/1,每个点存0/1,一条边真正的值为边的值异或两端的值。考虑树剖维护。
1、简单线段树操作
2、将a到b的点颜色取反!!!orz
对于路径上的边,两端都被取反相当于不变。对于相邻的边,有一个端点被改变相当于取反。u=lca(a,b)。对于u到父亲的边,a重孩子到a的边,b重孩子到b的边,暴力改一下颜色就好。但这样不是只正确维护了两端在同一重链的边么?对于跨重链的边,我们在询问时暴力查两端的点的颜色计算。由于每次询问最多跨log次重链,所以不会破坏复杂度。
3、还用说么。。
引入点颜色的想法太妙了。。