首页 > 代码库 > 【学习笔记】dsu on tree

【学习笔记】dsu on tree

我也不知道为啥这要起这名,完完全全没看到并查集的影子啊……

实际上原理就是一个树上的启发式合并。

特点是可以在$O(nlogn)$的时间复杂度内完成对无修改的子树的统计,复杂度优于莫队算法。

局限性也很明显:1.不能支持修改  2.只能支持子树统计,不能链上统计。(链上统计你不能直接树剖吗?)

那么它是怎么实现的呢?首先有一个例子:
树上每个节点都有一个颜色(那么一定是蓝色)

求每个节点的子树上有多少颜色为k的节点。(每个节点的k不一定相同)

$O(n^2)$的算法非常好想,以每个点为起点dfs一下就没了。

当然也有不那么暴力的做法,dfs序一下再主席树或者莫队随便搞搞也行。

那么我们先看看暴力是怎么做的。

每次统计x节点前,暴力将x的子树的贡献加入,统计结束后,再暴力删除贡献,消除影响。

我们发现这有很多无用的删除操作,考虑优化?

那么我们怎么用dsu上树优雅的解决这个问题呢?我们想到了树链剖分(轻重链剖分)。

具体的做法是,我们先统计一个点的轻儿子,再把它的影响消除。再统计重儿子,此时不必消除影响。

为了完成统计,最后再统计一遍轻儿子。

可以这么考虑:只有dfs到轻边时,才会将轻边的子树中合并到上一级的重链,

树链剖分将一棵树分割成了不超过$logn$条重链。
每一个节点最多向上合并$logn$次,单次修改复杂度$O(1)$。
所以整体复杂度是$O(nlogn)$的。

所以大概的模版是这样的:

 1 void dfs2(int u,int f,int k){
 2     for(int i=head[u];i;i=G[i].next){
 3         int v=G[i].v;if(v==f||v==wson[u])continue;
 4         dfs2(v,u,0);
 5     }
 6     if(wson[u])dfs(wson[u],u,1),now=wson[u];
 7     calc(u,f,1);
 8     now=0;ans[u]=sum;
 9     if(k==0)calc(u,f,-1),sumv=0,maxv=0;
10 }

下面是两道烂大街的例题:

1. Lomsat gelral(cf600E)

n个点的有根树,以1为根,每个点有一种颜色。我们称一种颜色占领了一个子树当且仅当没有其他颜色在这个子树中出现得比它多。求占领每个子树的所有颜色之和。

就是刚才的裸题啊。

 1 #include<bits/stdc++.h>
 2 #define N 700010
 3 using namespace std;
 4 struct Edge{int u,v,next;}G[2*N];
 5 typedef long long ll;
 6 int n,c[N],val[N],size[N],wson[N],fa[N];
 7 ll ans[N];
 8 int head[4*N],tot=0;
 9 void addedge(int u,int v){
10     G[++tot].u=u;G[tot].v=v;G[tot].next=head[u];head[u]=tot;
11     G[++tot].u=v;G[tot].v=u;G[tot].next=head[v];head[v]=tot;
12 }
13 void dfs1(int u,int f=0){
14     size[u]=1;
15     for(int i=head[u];i;i=G[i].next){
16         int v=G[i].v;if(v==f)continue;
17         if(v==f)continue;
18         dfs1(v,u);
19         size[u]+=size[v];
20         if(size[v]>size[wson[u]])wson[u]=v;
21     }
22 }
23 bool vis[N];int maxv=0;ll sum=0;
24 void change(int u,int f,int k){
25     c[val[u]]+=k;
26     if(k>0&&c[val[u]]>=maxv){
27         if(c[val[u]]>maxv)sum=0,maxv=c[val[u]];
28         sum+=val[u];
29     }
30     for(int i=head[u];i;i=G[i].next){
31         int v=G[i].v;if(v==f||vis[v])continue;
32         change(v,u,k);
33     }
34 }
35 void dfs2(int u,int f=0,bool used=0){
36     for(int i=head[u];i;i=G[i].next){
37         int v=G[i].v;if(v==f||v==wson[u])continue;
38         dfs2(v,u);
39     }
40     if(wson[u])dfs2(wson[u],u,1),vis[wson[u]]=1;
41     change(u,f,1);ans[u]=sum;
42     if(wson[u])vis[wson[u]]=0;
43     if(!used)change(u,f,-1),maxv=sum=0;
44 }
45 inline int read(){
46     int f=1,x=0;char ch;
47     do{ch=getchar();if(ch==-)f=-1;}while(ch<0||ch>9);
48     do{x=x*10+ch-0;ch=getchar();}while(ch>=0&&ch<=9);
49     return f*x;
50 }
51 int main(){
52     n=read();
53     for(int i=1;i<=n;i++)val[i]=read();
54     for(int i=1;i<n;i++){
55         int u=read(),v=read();
56         addedge(u,v);
57     }
58     dfs1(1);dfs2(1);
59     for(int i=1;i<=n;i++)printf("%I64d ",ans[i]);
60 }

当然这题也有不这么做的做法,随便从cf上粘了一个,大家自行意会……

 1 #include<bits/stdc++.h>
 2 #define N 100005
 3 using namespace std;
 4 vector<int>a[N];map<int,int>S[N];
 5 int F[N],id[N],c[N],n,i,x,y;
 6 long long G[N],ans[N];
 7 void work(int x,int y,int color){
 8     if (y>F[x]) F[x]=y,G[x]=0;
 9     if (y==F[x]) G[x]+=color;
10 }
11 void Union(int &x,int y){
12     if (S[x].size()<S[y].size()) swap(x,y);
13     for (map<int,int>::iterator it=S[y].begin();it!=S[y].end();it++)
14         work(x,S[x][it->first]+=it->second,it->first);
15 }
16 void DFS(int x,int fa){
17     id[x]=x;S[x][c[x]]=1;
18     F[x]=1;G[x]=c[x];
19     for (int i=0,y;i<a[x].size();i++)
20         if ((y=a[x][i])!=fa)
21             DFS(y,x),Union(id[x],id[y]);
22     ans[x]=G[id[x]];
23 }
24 int main(){
25     scanf("%d",&n);
26     for (i=1;i<=n;i++)
27         scanf("%d",&c[i]);
28     for (i=1;i<n;i++)
29         scanf("%d%d",&x,&y),
30         a[x].push_back(y),
31         a[y].push_back(x);
32     DFS(1,0);
33     for (i=1;i<=n;i++)
34         printf("%I64d ",ans[i]);
35 }

例2: Arpa‘s letter-marked tree and Mehrdad‘s Dokhtar-kosh paths(CF741D)

这题也很显然,如果重排后能形成回文串,那么出现奇数次的字符应该少于2个(即最多1个)如果只有a~v的话考虑把每个字符当做一个二进制位,把一个点i到根的路径异或值记为s[i],那么我们就是要对于每个x在子树中找到a和b,使得s[a]^s[b]为0或2的次幂,且dep[a]+dep[b]-dep[lca]*2最大。

 1 #include<bits/stdc++.h>
 2 #define N 500005
 3 using namespace std;
 4 int size[N],head[4*N],tot=0,wson[N],s[N],f[20*N],ans[N],d[N],a[N];
 5 char c[N];
 6 int maxv,n,inf;
 7 struct Edge{int u,v,next;}G[2*N];
 8 void addedge(int u,int v){
 9     G[++tot].u=u;G[tot].v=v;G[tot].next=head[u];head[u]=tot;
10     //G[++tot].u=v;G[tot].v=u;G[tot].next=head[v];head[v]=tot;
11 }
12 void dfs1(int u,int fa){
13     size[u]=1;d[u]=d[fa]+1;
14     if(u!=1)s[u]=s[fa]^(1<<a[u]);
15     for(int i=head[u];i;i=G[i].next){
16         int v=G[i].v;
17         dfs1(v,u);
18         size[u]+=size[v];if(size[v]>size[wson[u]])wson[u]=v;
19     }
20 }
21 void calc(int rt,int u){
22     int now=s[u];
23     maxv=max(maxv,f[now]+d[u]-2*d[rt]);
24     if((s[u]^s[rt])==0)maxv=max(maxv,d[u]-d[rt]);
25     for(int i=0;i<22;i++){
26         now=(1<<i)^s[u];
27         maxv=max(maxv,f[now]+d[u]-2*d[rt]);
28         if((s[u]^s[rt])==(1<<i))maxv=max(maxv,d[u]-d[rt]);
29     }
30     for(int i=head[u];i;i=G[i].next){
31         int v=G[i].v;calc(rt,v);
32     }
33 }
34 void change(int u,int k){
35     if(k)f[s[u]]=max(f[s[u]],d[u]);
36     else f[s[u]]=inf;
37     for(int i=head[u];i;i=G[i].next)change(G[i].v,k);
38 }
39 void dfs2(int u,int k){
40     for(int i=head[u];i;i=G[i].next){
41         int v=G[i].v;if(v==wson[u])continue;
42         dfs2(v,0);
43     }
44     if(wson[u])dfs2(wson[u],1);
45     maxv=0;int now=s[u];
46     maxv=max(maxv,f[now]-d[u]);
47     for(int i=0;i<22;i++){
48         now=(1<<i)^s[u];
49         maxv=max(maxv,f[now]-d[u]);
50     }
51     for(int i=head[u];i;i=G[i].next){
52         int v=G[i].v;if(v==wson[u])continue;
53         calc(u,v);change(v,1);
54     }
55     ans[u]=maxv;
56     if(!k){
57         for(int i=head[u];i;i=G[i].next)change(G[i].v,0);
58         f[s[u]]=inf;
59     }else f[s[u]]=max(f[s[u]],d[u]);
60 }
61 void erase(int u){
62     for(int i=head[u];i;i=G[i].next){
63         int v=G[i].v;erase(v);
64         ans[u]=max(ans[u],ans[v]);
65     }
66 }
67 int main(){
68     scanf("%d",&n);
69     for(int i=2;i<=n;i++){
70         int u;scanf("%d %c\n",&u,&c[i]);
71         addedge(u,i);a[i]=c[i]-a;
72     }
73     dfs1(1,0);
74     memset(f,128,sizeof(f));inf=f[0];dfs2(1,0);
75     erase(1);
76     for (int i=1;i<=n;++i)printf("%d%c",ans[i]," \n"[i==n]);
77 }

大概是这样。

参考:

http://blog.csdn.net/qq_35392050/article/details/64537364

http://www.cnblogs.com/zzqsblog/p/6146916.html

【学习笔记】dsu on tree