【题目链接】 http://acm.hdu.edu.cn/showproblem.php?pid=5893
【题目大意】
给出一棵树,每条边上都有一个边权,现在有两个操作,操作一要求将x到y路径上所有边更改为c权值,操作二要求查询x到y的路径上有几段连续的权值相同的。
【题解】
首先由于是边权,所以把所有边的存下来,做一遍剖分,将权值保存在每条边深度较深的点上,作为点权,用区间合并线段树维护区间内的线段段数,沿链修改的时候注意剖分出的区间的起点是不更新的,因为边权变成点权之后链修改的LCA是不修改的。查询的时候由于边权转点权之后点权位置的特殊性,我们每次在查询a到b之间的答案的时候,首先求出两者的LCA,同时求出LCA到a和b之间的第一个点,求分别求出a和b与其第二root之间的答案,再判断一下交接处的情况,就能计算出答案。
【代码】
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1000000;
int tot,x,d[N],num[N],ed=0,u,w,n,m,i,v[N],vis[N],f[N],g[N],nxt[N],size[N],son[N],st[N],en[N],dfn,top[N],t;char ch;
void add(int x,int y){v[++ed]=y;nxt[ed]=g[x];g[x]=ed;}
void dfs(int x){
size[x]=1;
for(int i=g[x];i;i=nxt[i])if(v[i]!=f[x]){
f[v[i]]=x,d[v[i]]=d[x]+1;
dfs(v[i]),size[x]+=size[v[i]];
if(size[v[i]]>size[son[x]])son[x]=v[i];
}
}
void dfs2(int x,int y){
if(x==-1)return;
st[x]=++dfn;top[x]=y;
if(son[x])dfs2(son[x],y);
for(int i=g[x];i;i=nxt[i])if(v[i]!=son[x]&&v[i]!=f[x])dfs2(v[i],v[i]);
en[x]=dfn;
}
int T[N<<2],mark[N<<2],cl[N<<2],cr[N<<2],L,R;
void up(int x){
T[x]=T[x<<1]+T[x<<1|1]-(cr[x<<1]==cl[x<<1|1]);
cl[x]=cl[x<<1];
cr[x]=cr[x<<1|1];
}
void pushdown(int x,int l,int r){
if(l==r)return;
if(mark[x]!=-1){
mark[x<<1]=mark[x<<1|1]=mark[x];
cl[x<<1]=cl[x<<1|1]=mark[x];
cr[x<<1]=cr[x<<1|1]=mark[x];
T[x<<1]=T[x<<1|1]=1;
mark[x]=-1;
}
}
void update(int x,int l,int r,int c){
pushdown(x,l,r);
if(L<=l&&r<=R){T[x]=1;mark[x]=cl[x]=cr[x]=c;return;}
int mid=(l+r)/2;
if(L<=mid)update(x<<1,l,mid,c);
if(mid<R)update(x<<1|1,mid+1,r,c);
up(x);
}
void update(int l,int r,int c){
if(l>r)return;
L=l;R=r; update(1,1,n,c);
}
int query(int x,int l,int r){
pushdown(x,l,r);
if(L<=l&&r<=R)return T[x];
int mid=(l+r)/2,ret=0;
if(L<=mid)ret+=query(x<<1,l,mid);
if(mid<R)ret+=query(x<<1|1,mid+1,r);
if(L<=mid&&mid<R)ret-=(cr[x<<1]==cl[x<<1|1]);
return ret;
}
int query(int l,int r){L=l;R=r;return query(1,1,n);}
int color(int x,int l,int r,int f){
if(l==r)return cl[x];
pushdown(x,l,r);
int mid=(l+r)/2;
if(f<=mid)return color(x<<1,l,mid,f);
return color(x<<1|1,mid+1,r,f);
}
int query(int l){return color(1,1,n,l);}
void chain(int x,int y,int c){
for(;top[x]!=top[y];x=f[top[x]]){
if(d[top[x]]<d[top[y]]){int z=x;x=y;y=z;}
update(st[top[x]],st[x],c);
}if(d[x]<d[y]){int z=x;x=y;y=z;}
update(st[y]+1,st[x],c);
}
int find(int x,int y){
int ret=0;
for(;top[x]!=top[y];x=f[top[x]]){
if(d[top[x]]<d[top[y]]){int z=x;x=y;y=z;}
ret+=query(st[top[x]],st[x]);
ret-=(query(st[top[x]])==query(st[f[top[x]]]));
}if(d[x]<d[y]){int z=x;x=y;y=z;}
ret+=query(st[y],st[x]);
return ret;
}
int lca(int x,int y){
for(;top[x]!=top[y];x=f[top[x]])if(d[top[x]]<d[top[y]]){int z=x;x=y;y=z;}
return d[x]<d[y]?x:y;
}
int lca2(int x,int y){
int t;
while(top[x]!=top[y])t=top[y],y=f[top[y]];
return x==y?t:son[x];
}
void init(){
for(int i=0;i<n*4;i++)T[i]=1,mark[i]=-1;
memset(g,dfn=ed=0,sizeof(g));
memset(v,0,sizeof(v));
memset(nxt,0,sizeof(nxt));
memset(son,-1,sizeof(son));
}
int cas;
int e[N][3];
int main(){
while(~scanf("%d%d",&n,&m)){
init();
for(int i=0;i<n-1;i++){
scanf("%d%d%d",&e[i][0],&e[i][1],&e[i][2]);
add(e[i][0],e[i][1]);
add(e[i][1],e[i][0]);
}dfs(1);dfs2(1,1);
for(int i=0;i<n-1;i++){
if(d[e[i][0]]>d[e[i][1]])swap(e[i][0],e[i][1]);
update(st[e[i][1]],st[e[i][1]],e[i][2]);
}char op[10]; int a,b,c;
//for(int i=1;i<=n;i++)printf("%d
",st[i]);
while(m--){
scanf("%s",op);
scanf("%d%d",&a,&b);
if(op[0]=='Q'){
c=lca(a,b);
int fa=lca2(c,a);
int fb=lca2(c,b);
//printf("%d %d %d
",c,fa,fb);
//printf("%d %d
",query(st[fa]),query(st[fb]));
if(a==b)puts("0");
else if(c==a)printf("%d
",find(b,fb));
else if(c==b)printf("%d
",find(a,fa));
else printf("%d
",find(a,fa)+find(b,fb)-(query(st[fa])==query(st[fb])));
}else{
scanf("%d",&c);
chain(a,b,c);
}
}
}return 0;
}