【bzoj3307】雨天的尾巴 线段树+树链剖分

考虑如果在序列上的话，每次只需要在区间左端点+1，右端点-1，最后用线段树一起统计即可。

在树上的话，就用树链剖分转换成序列，每一次操作，对应修改O(log n)条重链上的标记，最后用线段树来统计。

#include<cstdio>#include<cstring>#include<cstdlib>#include<cmath>#include<algorithm>#include<iostream>#include<vector>#include<map>#define maxn 200100using namespace std;struct yts{int l,r;int mx,tag;}t[4*maxn];int head[maxn],st[maxn],to[maxn],next[maxn],seq[maxn],rank[maxn],fa[19][maxn],size[maxn],dd[maxn];int n,m,num,tot,T,cnt;int e[maxn],pos[maxn],ans[maxn],dep[maxn];bool vis[maxn];vector<pair<int,int> > v[maxn];map<int,int> mp;void addedge(int x,int y){num++;to[num]=y;next[num]=st[x];st[x]=num;}void dfs(int x){size[x]=1;e[++tot]=x;for (int p=st[x];p;p=next[p])  if (to[p]!=fa[0][x])  {  fa[0][to[p]]=x;  dep[to[p]]=dep[x]+1;  dfs(to[p]);  size[x]+=size[to[p]];  if (size[to[p]]>size[dd[x]]) dd[x]=to[p];  }}int go_up(int x,int d){for (int i=0;i<=17;i++)  if (d&(1<<i)) x=fa[i][x];return x;}int LCA(int x,int y){if (dep[x]>dep[y]) x=go_up(x,dep[x]-dep[y]);else y=go_up(y,dep[y]-dep[x]);if (x==y) return x;for (int i=17;i>=0;i--)  if (fa[i][x]!=fa[i][y]) x=fa[i][x],y=fa[i][y];return fa[0][x];}void add(int x,int a,int b){v[x].push_back(make_pair(a,b));}void update(int i){t[i].mx=0;t[i].tag=0;if (t[i<<1].mx>t[i].mx) t[i].mx=t[i<<1].mx,t[i].tag=t[i<<1].tag;if (t[i<<1|1].mx>t[i].mx) t[i].mx=t[i<<1|1].mx,t[i].tag=t[i<<1|1].tag;else if (t[i<<1|1].mx==t[i].mx && t[i].tag>t[i<<1|1].tag) t[i].tag=t[i<<1|1].tag; }void build(int i,int l,int r){t[i].l=l;t[i].r=r;t[i].mx=0;if (l==r) {t[i].tag=pos[l];return;}int mid=(l+r)/2;build(i*2,l,mid);build(i*2+1,mid+1,r);}void modify(int i,int x,int d){if (t[i].l==t[i].r) {t[i].mx+=d;return;}int mid=(t[i].l+t[i].r)/2;if (x<=mid) modify(i*2,x,d);if (mid<x) modify(i*2+1,x,d);update(i);}int main(){scanf("%d%d",&n,&T);for (int i=1;i<n;i++){int x,y;scanf("%d%d",&x,&y);addedge(x,y);addedge(y,x);}dfs(1);int qwer=0;for (int i=1;i<=n;i++)  if (!vis[e[i]])  {  int k=e[i];  while (k)  {  seq[++qwer]=k;vis[k]=1;head[k]=e[i];k=dd[k];  }  }for (int i=1;i<=n;i++) rank[seq[i]]=i;for (int j=1;j<=17;j++)  for (int i=1;i<=n;i++)    fa[j][i]=fa[j-1][fa[j-1][i]];while (T--){int x,y,z;scanf("%d%d%d",&x,&y,&z);if (dep[x]<dep[y]) swap(x,y);if (!mp[z]) mp[z]=++cnt,pos[cnt]=z;int lca=LCA(x,y);z=mp[z];while (dep[head[x]]>dep[lca]){add(rank[head[x]],z,1);add(rank[x]+1,z,-1);x=fa[0][head[x]];}add(rank[lca],z,1);add(rank[x]+1,z,-1);if (y!=lca){int w=go_up(y,dep[y]-dep[lca]-1);while (dep[head[y]]>dep[w]){add(rank[head[y]],z,1);add(rank[y]+1,z,-1);y=fa[0][head[y]];}add(rank[w],z,1);add(rank[y]+1,z,-1);}}build(1,1,cnt);for (int i=1;i<=n;i++){for (int j=0;j<v[i].size();j++) modify(1,v[i][j].first,v[i][j].second);ans[seq[i]]=t[1].tag;}for (int i=1;i<=n;i++) printf("%d\n",ans[i]);return 0;}