标签：保留 节点 data print str return head log name
今天介绍一个神仙算法：Dsu On Tree[ 树上启发式合并 ]
这个算法用于离线处理询问子树信息，而且很好写。
但是在你没有理解它之前，这是个很鬼畜的算法。
理解后你才能真心感到它的美妙之处。
关键是它是有着媲美线段树合并的时间复杂度的“暴力”算法。
这里说一件事，我学这个东西时找了很多篇博客，它们无一例外地给出了这样一个流程：
1. 先统计一个节点所有的轻儿子 然后删除它的答案
2. 再统计这个节点的重儿子 保留他的答案
3. 再算一遍所有轻儿子 加到答案中上传
我当时就看的很懵逼，算一遍所有轻儿子，删掉，再算一遍，这不闲的？
直接统计它的重儿子再算轻儿子不就好了？很疑惑，问了身边很多人也都觉得迷惑。
人类迷惑行为大赏.jpg
后面我搞懂了，为了不让其他学习dsu on tree的人也觉得迷惑，我就写了这一篇博客。
在这里非常感谢洛谷两个dalao的帮助，现在理解了这个东西。
我们每次进入一棵子树计算答案时，都要把计算上一棵子树的数据清除。
为什么，如果我们带着上次计算后的结果去计算新子树，答案肯定是不对的。
但是，我们最后一棵子树不需要清除，因为我们不用再进入新子树了（没了）。
那我们再回到上面说的，为什么一开始要算一遍轻儿子？
从最纯粹的暴力开始，我们有两个函数dfs1和dfs2，dfs1函数作为主体函数，dfs2作为辅助函数。
先dfs1到每一个点，dfs1它的后代，计算后代的信息，再dfs2它的后代，计算自己的答案。
也就是说，开始算轻儿子是要把它后代的信息计算出来，而不是理解为之前提到那些博客里面的算出来答案后“删除答案”。删除答案是为了不让计算的数据冲突。
按照dalao说的，保留重儿子的信息可以优化复杂度。从原先暴力的O(n^2)优化到O(nlogn)。
所以为了不用清除重儿子的信息，先dfs1轻儿子，再dfs1重儿子，最后dfs2轻儿子更新自己的答案。
如果一开始没有dfs1轻儿子，我们就没有得到后代的信息，所谓“删除”答案，其实是从统计信息的数组把dfs1轻儿子时存进去的用于计算的“缓存”清理了。
下面给出代码：
#include<bits/stdc++.h> #define N 100010 using namespace std; inline int read(){ int data=0,w=1;char ch=0; while(ch!=‘-‘ && (ch<‘0‘||ch>‘9‘))ch=getchar(); if(ch==‘-‘)w=-1,ch=getchar(); while(ch>=‘0‘ && ch<=‘9‘)data=data*10+ch-‘0‘,ch=getchar(); return data*w; } struct Edge{ int nxt,to; #define nxt(x) e[x].nxt #define to(x) e[x].to }e[N<<1]; int head[N],tot; inline void addedge(int f,int t){ nxt(++tot)=head[f];to(tot)=t;head[f]=tot; } int cnt[N],siz[N],son[N],c[N],max_val,n,child; long long sum,ans[N]; void add(int x,int f,int val){ cnt[c[x]]+=val; if(cnt[c[x]]>max_val)max_val=cnt[c[x]],sum=c[x]; else if(cnt[c[x]]==max_val)sum+=1LL*c[x]; for(int i=head[x];i;i=nxt(i)){ int y=to(i); if(y==f||y==child)continue; add(y,x,val); } } void dfs1(int x,int f){//重链剖分 siz[x]=1;int maxson=-1; for(int i=head[x];i;i=nxt(i)){ int y=to(i); if(y==f)continue; dfs1(y,x); siz[x]+=siz[y]; if(siz[y]>maxson){ maxson=siz[y];son[x]=y; } } } void dfs2(int x,int f,int opt){//opt为0表示统计后的答案要删掉，opt为1则不用删 for(int i=head[x];i;i=nxt(i)){ int y=to(i); if(y==f)continue; if(y!=son[x])dfs2(y,x,0); }if(son[x])dfs2(son[x],x,1),child=son[x]; add(x,f,1);child=0; ans[x]=sum; if(!opt)add(x,f,-1),sum=0,max_val=0; } int main(){ n=read(); for(int i=1;i<=n;i++)c[i]=read(); for(int i=1;i<n;i++){ int x=read(),y=read(); addedge(x,y);addedge(y,x); } dfs1(1,0);dfs2(1,0,0); for(int i=1;i<=n;i++) printf("%lld ",ans[i]); return 0; }
标签：保留 节点 data print str return head log name
原文地址：https://www.cnblogs.com/light-house/p/11779076.html
