题意:正常人都看得懂吧
给您一颗树,每个节点有个初始值。
现在支持以下两种操作:
1. C i x(0<=x<2^31) 表示将i节点的值改为x。
2. Q i j x(0<=x<2^31) 表示询问i节点到j节点的路径上有多少个值为x的节点。
题解:
树链剖分+线段树。先把所有值离散化,然后开许多棵线段树,记录每个值的个数。因为空间过不去,所以我用了许多指针。真tm不习惯
代码:
#include<bits/stdc++.h>
using namespace std;
struct tree
{
int l,r,c;
tree *lc,*rc;
tree(int _l,int _r)
{
l=_l;
r=_r;
c=0;
lc=rc=NULL;
}
}*root[300010];
struct pnt
{
int f,s,sz,en,tp,dp;
}p[100010];
struct edge
{
int x,y,n;
}e[200010];
struct query
{
char s[2];
int x,y,z;
}q[200010];
int n,m,nn=0,a[100010],fst[100010],num=0;
map<int,int>lsh;
void ins(int x,int y)
{
e[++num]={x,y,fst[x]};
fst[x]=num;
}
void dfs1(int x,int f)
{
p[x].f=f;
p[x].dp=p[f].dp+1;
p[x].sz=1;
for(int i=fst[x];i;i=e[i].n)
{
int y=e[i].y;
if(y==f)
continue;
dfs1(y,x);
if(p[y].sz>p[p[x].s].sz)
p[x].s=y;
p[x].sz+=p[y].sz;
}
}
void dfs2(int x,int tp)
{
p[x].tp=tp;
p[x].en=++nn;
if(p[x].s)
{
dfs2(p[x].s,tp);
for(int i=fst[x];i;i=e[i].n)
{
int y=e[i].y;
if(y==p[x].f||y==p[x].s)
continue;
dfs2(y,y);
}
}
}
bool del(tree* i,int x)
{
if(i==NULL)
return 1;
// printf("del %d %d %d\n",i->l,i->r,x);
if(i->l==i->r)
{
delete i;
return 1;
}
int md=i->l+i->r>>1;
if(x<=md)
{
if(del(i->lc,x))
i->lc=NULL;
}
else
{
if(del(i->rc,x))
i->rc=NULL;
}
// i->c=(i->lc!=NULL?i->lc->c:0)+(i->rc!=NULL?i->rc->c:0);
i->c--;
if(i->c==0)
{
delete i;
return 1;
}
return 0;
}
void add(tree* i,int x)
{
// printf("add %d %d %d\n",i->l,i->r,x);
// system("pause");
if(i->l==i->r)
{
i->c=1;
return;
}
int md=i->l+i->r>>1;
tree *lc=i->lc,*rc=i->rc;
if(x<=md)
{
if(lc==NULL)
i->lc=lc=new tree(i->l,md);
add(lc,x);
}
else
{
if(rc==NULL)
i->rc=rc=new tree(md+1,i->r);
add(rc,x);
}
i->c=(lc!=NULL?lc->c:0)+(rc!=NULL?rc->c:0);
}
int get(tree* i,int l,int r)
{
if(i->l==l&&i->r==r)
{
return i->c;
}
int md=i->l+i->r>>1;
tree *lc=i->lc,*rc=i->rc;
if(r<=md)
{
return lc==NULL?0:get(lc,l,r);
}
else if(l>md)
{
return rc==NULL?0:get(rc,l,r);
}
else
{
return (lc==NULL?0:get(lc,l,md))+(rc==NULL?0:get(rc,md+1,r));
}
}
int slv(int x,int y,int c)
{
int ans=0;
while(p[x].tp!=p[y].tp)
{
if(p[p[x].tp].dp<p[p[y].tp].dp)
swap(x,y);
int tx=p[x].tp,ty=p[y].tp;
ans+=get(root[c],p[tx].en,p[x].en);
x=p[tx].f;
}
if(p[x].dp<p[y].dp)
swap(x,y);
ans+=get(root[c],p[y].en,p[x].en);
return ans;
}
void print(tree *x)
{
if(x==NULL)
return;
printf("l:%d r:%d c:%d\n",x->l,x->r,x->c);
print(x->lc);
print(x->rc);
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
lsh[a[i]]=1;
}
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
ins(x,y);
ins(y,x);
}
for(int i=1;i<=m;i++)
{
scanf("%s%d%d",q[i].s,&q[i].x,&q[i].y);
if(q[i].s[0]=='Q')
{
scanf("%d",&q[i].z);
lsh[q[i].z]=1;
}
else
lsh[q[i].y]=1;
}
int cnt=0;
for(map<int,int>::iterator it=lsh.begin();it!=lsh.end();it++)
{
it->second=cnt++;
// printf("lsh[%d]=%d\n",it->first,it->second);
}
dfs1(1,0);
dfs2(1,1);/*
puts("en:");
for(int i=1;i<=n;i++)
printf("%d ",p[i].en);
puts("");*/
for(int i=0;i<cnt;i++)
root[i]=new tree(1,nn);
for(int i=1;i<=n;i++)
{
a[i]=lsh[a[i]];
add(root[a[i]],p[i].en);
}/*
for(int i=0;i<cnt;i++)
{
printf("tree:%d\n",i);
print(root[i]);
}*/
for(int i=1;i<=m;i++)
{
if(q[i].s[0]=='C')
{
q[i].y=lsh[q[i].y];
if(del(root[a[q[i].x]],p[q[i].x].en))
root[a[q[i].x]]=new tree(1,nn);
add(root[q[i].y],p[q[i].x].en);
a[q[i].x]=q[i].y;
}
else
{
q[i].z=lsh[q[i].z];
printf("%d\n",slv(q[i].x,q[i].y,q[i].z));
}/*
for(int i=0;i<cnt;i++)
{
printf("tree:%d\n",i);
print(root[i]);
}*/
}
}