Description
给定一棵有n个节点的无根树和m个操作,操作有2类:
1、将节点a到节点b路径上所有点都染成颜色c;
2、询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段),如“112221”由3段组成:“11”、“222”和“1”。
请你写一个程序依次完成这m个操作。
Input
第一行包含2个整数n和m,分别表示节点数和操作数;
第二行包含n个正整数表示n个节点的初始颜色
下面 行每行包含两个整数x和y,表示x和y之间有一条无向边。
下面 行每行描述一个操作:
“C a b c”表示这是一个染色操作,把节点a到节点b路径上所有点(包括a和b)都染成颜色c;
“Q a b”表示这是一个询问操作,询问节点a到节点b(包括a和b)路径上的颜色段数量。
Output
对于每个询问操作,输出一行答案。
Sample Input
6 5
2 2 1 2 1 1
1 2
1 3
2 4
2 5
2 6
Q 3 5
C 2 1 1
Q 3 5
C 5 1 2
Q 3 5
Sample Output
3
1
2
思路
树链剖分+线段树就好了。
第一个dfs找size、deep;第二个dfs递归找重儿子,然后连在一起建立种链。lca用来算两个节点之间的路径,然后就是线段树上的事情了。
代码
好像WA了。。。
#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=100000+5;
int n,m,num,sz,head[N],deep[N],son[N];
int belong[N],pl[N],v[N],ft[N][18];
bool vis[N];
struct seg
{
int l,r,lc,rc,s,tag;
};
seg tree[4*N];
struct edge
{
int u,next;
};
edge ed[2*N];
void insert(int u,int v)
{
ed[++num].u=v;
ed[num].next=head[u];
head[u]=num;
ed[++num].u=u;
ed[num].next=head[v];
head[v]=num;
}
void dfs1(int x)
{
vis[x]=son[x]=1;
for (int i=1;i<=17;i++)
{
if (deep[x]<(1<<i)) break;
ft[x][i]=ft[ft[x][i-1]][i-1];
}
for (int i=head[x];i;i=ed[i].next)
{
if (vis[ed[i].u]) continue;
deep[ed[i].u]=deep[x]+1;
ft[ed[i].u][0]=x;
dfs1(ed[i].u);
son[x]+=son[ed[i].u];
}
}
void dfs2(int x,int chain)
{
int k=0;
pl[x]=++sz;belong[x]=chain;
for (int i=head[x];i;i=ed[i].next)
if (deep[ed[i].u]>deep[x]&&son[k]<son[ed[i].u]) k=ed[i].u;
if (!k) return ;
dfs2(k,chain);
for (int i=head[x];i;i=ed[i].next)
if (deep[ed[i].u]>deep[x]&&k!=ed[i].u)
dfs2(ed[i].u,ed[i].u);
}
int lca(int x,int y)
{
if (deep[x]<deep[y]) swap(x,y);
int t=deep[x]-deep[y];
for (int i=0;i<=17;i++)
if (t&(1<<i)) x=ft[x][i];
for (int i=17;i>=0;i--)
if (ft[x][i]!=ft[y][i]) {x=ft[x][i];y=ft[y][i];}
if (x==y) return x;
return ft[x][0];
}
void build(int k,int lf,int rt)
{
tree[k].l=lf;
tree[k].r=rt;
tree[k].s=1;
tree[k].tag=-1;
if (lf==rt) return ;
int mid=(lf+rt)/2;
build(k*2,lf,mid);
build(k*2|1,mid+1,rt);
}
void pushup(int k)
{
tree[k].lc=tree[k*2].lc;
tree[k].rc=tree[k*2+1].rc;
if (tree[k*2].rc^tree[k*2+1].lc)
tree[k].s=tree[k*2].s+tree[k*2+1].s-1;
}
void pushdown(int k)
{
int tmp=tree[k].tag;
tree[k].tag=-1;
if (tmp==-1||tree[k].l==tree[k].r) return ;
tree[k*2].s=tree[k*2+1].s=1;
tree[k*2].tag=tree[k*2+1].tag=tmp;
tree[k*2].lc=tree[k*2+1].rc=tmp;
tree[k*2+1].lc=tree[k*2+1].rc=tmp;
}
void change(int k,int x,int y,int c)
{
pushdown(k);
int lf=tree[k].l,rt=tree[k].r;
if (lf==x&&rt==y)
{
tree[k].lc=tree[k].rc=c;
tree[k].s=1;
tree[k].tag=c;
return ;
}
int mid=(lf+rt)/2;
if (mid>=y) change(k<<1,x,y,c);
else if (mid<x) change(k<<1|1,x,y,c);
else
{
change(k*2,x,mid,c);
change(k*2+1,mid+1,y,c);
}
pushup(k);
}
int ask(int k,int x,int y)
{
pushdown(k);
int l=tree[k].l,r=tree[k].r;
if(l==x&&r==y)return tree[k].s;
int mid=(l+r)>>1;
if(mid>=y)return ask(k<<1,x,y);
else if(mid<x)return ask(k<<1|1,x,y);
else
{
int tmp=1;
if(tree[k<<1].rc^tree[k<<1|1].lc)tmp=0;
return ask(k<<1,x,mid)+ask(k<<1|1,mid+1,y)-tmp;
}
}
int getc(int k,int x)
{
pushdown(k);
int l=tree[k].l,r=tree[k].r;
if(l==r)return tree[k].lc;
int mid=(l+r)>>1;
if(x<=mid)return getc(k<<1,x);
else return getc(k<<1|1,x);
}
int solvesum(int x,int f)
{
int sum=0;
while(belong[x]!=belong[f])
{
sum+=ask(1,pl[belong[x]],pl[x]);
if (getc(1,pl[belong[x]])==getc(1,pl[ft[belong[x]][0]]))
sum--;
x=ft[belong[x]][0];
}
sum+=ask(1,pl[f],pl[x]);
return sum;
}
void solvechange(int x,int f,int c)
{
while(belong[x]!=belong[f])
{
change(1,pl[belong[x]],pl[x],c);
x=ft[belong[x]][0];
}
change(1,pl[f],pl[x],c);
}
void solve()
{
int a,b,c;
dfs1(1);
dfs2(1,1);
build(1,1,n);
for(int i=1;i<=n;i++)
change(1,pl[i],pl[i],v[i]);
for(int i=1;i<=m;i++)
{
char ch[10];
scanf("%s",ch);
if(ch[0]=='Q')
{
scanf("%d%d",&a,&b);
int t=lca(a,b);
printf("%d\n",solvesum(a,t)+solvesum(b,t)-1);
}
else
{
scanf("%d%d%d",&a,&b,&c);
int t=lca(a,b);
solvechange(a,t,c);solvechange(b,t,c);
}
}
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&v[i]);
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
insert(x,y);
}
solve();
return 0;
}