[BZOJ 2243][SDOI2011] 染色 树链剖分 模板题

题目传送门：【BZOJ 2243】

---

题目大意：给定一棵有 n 个节点的无根树和 m 个操作，操作有2类：
1、将节点a到节点b路径上所有点都染成颜色c；
2、询问节点a到节点b路径上的颜色段数量（连续相同颜色被认为是同一段），如“112221”由3段组成：“11”、“222”和“1”。

---

题目分析：
（说好的是一道模板题，结果花了本人 4 个小时修改之后的线段树）
如题，通常来讲，询问一棵树上的节点信息这样的题，并且包含着“询问两个不同节点的路径上的信息”时，基本上就可以确定是树链剖分的题了。

所以这道题就是道实实在在的树链剖分模板。

但是！

这道题比较坑的地方在于不同颜色的合并，合并时我们用线段树来进行操作（当然是树链剖分之后的线段树），这时就要注意重复的问题。如果左子树最右边的点的颜色和右子树最左边的点颜色相同，那么，合并之后的颜色区块数量就要-1。所以我们需要单独记录下左右两个端点的颜色信息。

同样还有一个细节：颜色的区间为 [ 0 , 10$^9$ ]，所以区间修改的标记在初始化时应为 -1。

至于最后一个问题嘛……有人问为什么一直 RE……
很可能是你的输入操作 ‘C’ 和 ‘Q’ 这个地方有问题……
用字符串输入就行啦

下面附上代码（附注释）：

[cpp] view plain copy

print ?

1. #include<cstdio>  
2. #include<algorithm>  
3.   
4. #define ls (nd<<1)  
5. #define rs (nd<<1|1)  
6. #define Root 1  
7. using namespace std;  
8. const int MX=100005;  
9.   
10. struct Edge{  
11.     int to,next;  
12. }edge[MX*2];  
13. int n,m,now=0,_index=0;  
14. int line[MX],head[MX];  
15. int depth[MX],size[MX],fa[MX],son[MX],top[MX],in[MX],seq[MX];  
16. struct SegTree{  
17.     int size,flag;          //size:不同颜色块的数量 flag:新颜色   
18.     int l,r;                //l,r:两端颜色   
19. }seg[MX*4];  
20.   
21. /*————–  ↓树链剖分↓  ————–*/  
22. inline void adde(int u,int v){  
23.     edge[++now].to=v;  
24.     edge[now].next=head[u];  
25.     head[u]=now;  
26. }  
27. void dfs1(int u){  
28.     size[u]=1;  
29.     for (int i=head[u];i;i=edge[i].next){  
30.         int v=edge[i].to;  
31.         if (fa[u]==v) continue;  
32.         fa[v]=u;  
33.         depth[v]=depth[u]+1;  
34.         dfs1(v);  
35.         if (size[v]>size[son[u]]) son[u]=v;  
36.         size[u]+=size[v];  
37.     }  
38. }  
39. void dfs2(int u,int tp){  
40.     top[u]=tp;  
41.     seq[++_index]=u;  
42.     in[u]=_index;  
43.     if (son[u]) dfs2(son[u],tp);  
44.     for (int i=head[u];i;i=edge[i].next){  
45.         int v=edge[i].to;  
46.         if (fa[u]==v || son[u]==v) continue;  
47.         dfs2(v,v);  
48.     }  
49. }  
50. /*————–   ↓线段树↓   ————–*/  
51. void build(int nd,int lf,int rt){  
52.     seg[nd].flag=-1;                        //flag可以为0，因此设为-1保证正确性   
53.     if (lf==rt){  
54.         seg[nd].size=1;  
55.         seg[nd].l=seg[nd].r=line[seq[lf]];  
56.     } else {  
57.         int mid=(lf+rt)>>1;  
58.         build(ls,lf,mid);  
59.         build(rs,mid+1,rt);  
60.         seg[nd].size=seg[ls].size+seg[rs].size-(seg[ls].r==seg[rs].l);  
61.         seg[nd].l=seg[ls].l;  
62.         seg[nd].r=seg[rs].r;  
63.     }  
64. }  
65. void pushdown(int nd){  
66.     if (seg[nd].flag!=-1){  
67.         seg[ls].flag=seg[rs].flag=seg[nd].flag;  
68.         seg[ls].l=seg[ls].r=seg[rs].l=seg[rs].r=seg[nd].flag;//两端都要改   
69.         seg[ls].size=seg[rs].size=1;  
70.         seg[nd].flag=-1;  
71.     }  
72. }  
73. void update(int nd){  
74.     seg[nd].size=seg[ls].size+seg[rs].size-(seg[ls].r==seg[rs].l);  
75.     seg[nd].l=seg[ls].l,seg[nd].r=seg[rs].r;  
76. }  
77. void modify(int nd,int lf,int rt,int L,int R,int val){  
78.     if (L>R) return;  
79.     if (L<=lf && rt<=R){  
80.         seg[nd].size=1;  
81.         seg[nd].l=seg[nd].r=seg[nd].flag=val;  
82.         return;  
83.     }  
84.     pushdown(nd);  
85.     int mid=(lf+rt)>>1;  
86.     if (L<=mid)  
87.         modify(ls,lf,mid,L,R,val);  
88.     if (R>mid)  
89.         modify(rs,mid+1,rt,L,R,val);  
90.     update(nd);  
91. }  
92. int query(int nd,int lf,int rt,int L,int R){//合并两条重链时要判重   
93.     if (L>R) return 0;  
94.     if (L<=lf && rt<=R)  
95.         return seg[nd].size;  
96.           
97.     pushdown(nd);  
98.     int mid=(lf+rt)>>1,ans=0;  
99.     if (L<=mid && R>mid)                  //合并要去重，所以有三种情况   
100.         ans+=query(ls,lf,mid,L,R)+query(rs,mid+1,rt,L,R)-(seg[ls].r==seg[rs].l);  
101.     else if (L<=mid)  
102.         ans+=query(ls,lf,mid,L,R);  
103.     else if (R>mid)  
104.         ans+=query(rs,mid+1,rt,L,R);  
105.     return ans;  
106. }  
107. int query_light(int nd,int lf,int rt,int v){//合并两条轻边时也要判重   
108.     if (lf==rt)  
109.         return seg[nd].l;  
110.           
111.     pushdown(nd);  
112.     int mid=(lf+rt)>>1;  
113.     if (v<=mid)  
114.         return query_light(ls,lf,mid,v);  
115.     else  
116.         return query_light(rs,mid+1,rt,v);  
117. }  
118. void modify(int u,int v,int color){  
119.     while (top[u]!=top[v]){  
120.         if (depth[top[u]]<depth[top[v]]) swap(u,v);  
121.         modify(Root,1,n,in[top[u]],in[u],color);  
122.         u=fa[top[u]];  
123.     }  
124.     if (depth[u]<depth[v]) swap(u,v);  
125.     modify(Root,1,n,in[v],in[u],color);  
126. }  
127. int query(int u,int v){  
128.     int ans=0;  
129.     while (top[u]!=top[v]){  
130.         if (depth[top[u]]<depth[top[v]]) swap(u,v);  
131.         ans+=query(Root,1,n,in[top[u]],in[u]);  
132.         //↓合并轻边的过程 ↓  
133.         if (query_light(Root,1,n,in[top[u]])==query_light(Root,1,n,in[fa[top[u]]]))  
134.             ans–;  
135.         u=fa[top[u]];  
136.     }  
137.     if (depth[u]<depth[v]) swap(u,v);  
138.     ans+=query(Root,1,n,in[v],in[u]);  
139.     return ans;  
140. }  
141.   
142. int main(){  
143.     char opt[2];  
144.     int a,b,color;  
145.     scanf(”%d%d”,&n,&m);  
146.     for (int i=1;i<=n;i++)  
147.         scanf(”%d”,&line[i]);  
148.     for (int i=1;i<n;i++){  
149.         scanf(”%d%d”,&a,&b);  
150.         adde(a,b),adde(b,a);  
151.     }  
152.     depth[1]=1,fa[1]=1;  
153.     dfs1(1);  
154.     dfs2(1,1);  
155.     build(Root,1,n);  
156.     for (int i=1;i<=m;i++){  
157.         scanf(”%s”,opt);  
158.         if (opt[0]==‘C’){  
159.             scanf(”%d%d%d”,&a,&b,&color);  
160.             modify(a,b,color);  
161.         }  
162.         if (opt[0]==‘Q’){  
163.             scanf(”%d%d”,&a,&b);  
164.             printf(”%d\n”,query(a,b));  
165.         }  
166.     }  
167.     return 0;  
168. }