[Ispc2009]Let there be rainbows!
Description
HY Star是一个处处充满和谐，人民安居乐业的星球，但是HYStar却没有被评上宇宙文明星球，很大程度上是因为星球的形象问题。
HY Star由N个国家组成,并且在一些国家之间修建了道路以方便交流。由于HY Star是一个和谐的
星球，因此任意两个国家有且仅有一条路径连接它们。而让HYStar备受诟病的便是他们修建的道路的颜色过于单调,全部是使用灰色。
经过了一年又一年的宇宙文明星球的落选,HY Star的首领BOSS决定改变这种状况，但是他并
不想重新修建新的道路，因为这太耗费人力物力,他决定将已有的道路重新涂色。在各方的讨论下，BOSS决定将现有的道路涂成彩虹的颜色。每一天,BOSS都会选择两个国家A,B和一种颜色C，并将A,B之间颜色不是C的道路的颜色都涂成C。
BOSS将涂色的具体计划预先告诉了颜料厂，以便有充足的时间来准备颜料。作为颜料厂的负责人,你需要知道每种颜色被使用了多少次。
Input
第一行,一个整数N,表示HY Star的国家的编号是1~N。
下面若干行，每行两个数A,B,表示A,B之间有道路直接连接。
下面一行,一个整数Q,表示计划的持续天数。
下面Q行,每行三个正整数,S,T,C(1<=S,T<=N,1<=C<=7),表示将S,T之间颜色不是C的道路的颜色都涂成C。
1 < = N,Q < = 200000
Output
共包含7行,第i行包含一个整数，表示颜色i的使用次数。
Sample Input:
4
1 2
2 3
3 4
3
1 4 1
2 4 2
1 3 1
Out put:
4
2
0
0
0
0
0
Solution:
本题可以用树链剖分来解决，线段树的维护方法有两种：
每个区间记录7个颜色的值，每个区间记录当前颜色值
第一种方法我们不在此讨论，我们来讲第二种方法
若当前区间的颜色为同一种，则记录这种颜色，否则为0
每次更新之前，就把这条区间长度减去原有的颜色数量，在把答案加到数组里去就行了
更新时就按照普通线段树来更新，要注意的是，本题细节很重要
Code:
#include<bits/stdc++.h>
#define N 2000001
#define inf 19260817
#define ls q<<1
#define rs q<<1|1
using namespace std;
int n,m,tl,ans[10];
int fa[N],top[N],son[N],sz[N],dep[N];
int cnt,seg[N],tag[N<<2];
int tot,head[N],nxt[N<<1],to[N<<1];
void ins(int x,int y){
nxt[++tot]=head[x];
to[tot]=y;head[x]=tot;
}
void update(int q){tag[q]=(tag[ls]==tag[rs])?tag[ls]:0;}
void pushdown(int q,int l,int r){
if(!tag[q])return ;
tag[ls]=tag[rs]=tag[q];
}
void change(int q,int l,int r,int L,int R,int v){
if(tag[q]==v)return ;
if(l>=L&&r<=R){
tag[q]=v;
return ;
}
pushdown(q,l,r);
int mid=(l+r)>>1;
if(mid>=L)change(ls,l,mid,L,R,v);
if(mid<R)change(rs,mid+1,r,L,R,v);
update(q);
}
int query(int q,int l,int r,int L,int R,int col){
if(l>=L&&r<=R&&tag[q]==col)return r-l+1;
if(tag[q]!=0&&tag[q]!=col)return 0;
if(l==r)return 0;
pushdown(q,l,r);
int mid=(l+r)>>1,res=0;
if(mid>=L)res+=query(ls,l,mid,L,R,col);
if(mid<R)res+=query(rs,mid+1,r,L,R,col);
return res;
}
void dfs1(int x,int f){
int mxsz=0;
fa[x]=f;dep[x]=dep[f]+1;sz[x]=1;
for(int i=head[x];i;i=nxt[i]){
int y=to[i];
if(y==f)continue;
dfs1(y,x);sz[x]+=sz[y];
if(sz[y]>mxsz)son[x]=y,mxsz=sz[y];
}
}
void dfs2(int x,int topx){
seg[x]=++cnt;top[x]=topx;
if(!son[x])return ;
if(son[x])dfs2(son[x],topx);
for(int i=head[x];i;i=nxt[i]){
int y=to[i];
if(y==fa[x]||y==son[x])continue;
dfs2(y,y);
}
}
int Query(int x,int y,int col){
int ans=0;
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
ans+=query(1,1,n,seg[top[x]],seg[x],col);
x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
ans+=query(1,1,n,seg[x]+1,seg[y],col);
return ans;
}
int Tot(int x,int y){
int t=0;
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
t+=seg[x]-seg[top[x]]+1;
x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
t+=seg[y]-seg[x];
return t;
}
void Change(int x,int y,int col){
int lst=Query(x,y,col);
int xx=x,yy=y;
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
change(1,1,n,seg[top[x]],seg[x],col);
x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
change(1,1,n,seg[x]+1,seg[y],col);
ans[col]+=Tot(xx,yy)-lst;
}
int read(){
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}
while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int main(){
n=read();
for(int i=1;i<n;i++){
int x=read(),y=read();
ins(x,y);ins(y,x);
}
m=read();dfs1(1,0);dfs2(1,1);
for(int i=1;i<=m;i++){
int x=read(),y=read(),col=read();
Change(x,y,col);
}
for(int i=1;i<=7;i++)printf("%d\n",ans[i]);
return 0;
}
