前言:
去年舒老师还健在的时候,这是ta问我的最后一道题
为了纪念ta,我打算在这个神圣的日子里完成一下这道题
想当年,我看到这道题时一秒口hu:树上莫队
(然而真的是树上莫队的裸题)
在不学新知识的情况下,试着用点分治搞一下
分析:
点分治——以我现在水平——认为只能解决子树内的路径计数之类的问题
要解决路径问题还是第一次见
显然我们应该是计算以每一个节点为LCA的点对之间路径的贡献
以舒老师的说法:
很明显,这种处理树上点对、nlogn可以接受且单次询问的问题,肯定是点分治跑不掉了
我们现在的问题就是:如何计算以某个结点为LCA的点对贡献
通过大量的画图和发呆
我发现了几个性质:
- 对于子树x,若其中i结点的颜色是独一无二的,那么i就可以与所有非该子树内的点形成点对,贡献就是size[root]-size[x]
- 对于树中的一点i,如果该点的颜色在该点到根这条链上是第一次出现,
那么对于这棵树的其他点j(j与i的lca为根结点),均能与i的子树(包括i)组成点对,i的颜色会对j的答案贡献size[i](我们暂且不考虑j到根的链上是否出现了i的颜色)
Q.
一开始我不是很理解第二个性质(这是舒老师传授的)
主要的疑惑就在于:要是i的子树内也有i的颜色怎么办呢
A.
但是注意原命题中有一个很重要的条件:i的颜色在该点到根这条链上是第一次出现
我们在继续dfs时候,如果又遇到了重复颜色是不会计算贡献的,所以这个性质是很优秀的
这道题中两个性质都有应用
整体的思路当然还是dfs分治
void solve(int now,int fa)
{
vis[now]=1;
doit(now,fa);
for (int i=st[now];i!=-1;i=way[i].nxt)
{
int v=way[i].y;
if (v!=fa&&!vis[v])
{
sz=size[v];
root=0;
getroot(v,0);
solve(root,0);
}
}
}
然而在分治的时候计算贡献就有一点点麻烦了:
dfs_1
什么都别啰嗦,一上来先dfs一遍
维护每个结点的子树大小
(因为每次我们solve的时候,root都有可能变化,所以每次都要维护size)
同时计算第二个性质中牵扯的贡献
这个贡献要怎么维护呢?
因为贡献牵扯到size,所以我们只能在回溯的时候记录
设置一个计数器cnt,按照dfs顺序进行对应颜色的计数
在回溯的时候,如果cnt=1,那么这个颜色对于当前这棵树的贡献就是size[i]
我们把每种颜色的贡献记录在一个数组中(color)
最后,因为我们要向上回溯,所以不要忘了消除影响
if (cnt[C[now]]==1)
{
sum+=size[now];
color[C[now]]+=size[now]; //C[now]对于当前这棵树的贡献
}
我们可以注意到代码中有一个变量:
sum
这个就是子树内每种颜色的贡献总和
枚举子树
void doit(int now,int fa)
{
dfs_1(now,fa);
ans[now]+=(ll)sum;
for (int i=st[now];i!=-1;i=way[i].nxt)
{
int v=way[i].y;
if (!vis[v]&&v!=fa)
{
cnt[C[now]]++;
color[C[now]]-=size[v]; sum-=size[v];
change(v,now,-1);
cnt[C[now]]--;
tt=size[now]-size[v];
dfs_2(v,now);
cnt[C[now]]++; //消除影响
color[C[now]]+=size[v]; sum+=size[v];
change(v,now,1);
cnt[C[now]]--;
}
}
sum=0;num=0;
clear(now,fa);
}
我们这样只处理了:以根结点为LCA的点对贡献,且两点一定在该子树内
但是对于那些“以根结点为LCA,但只有一个点在该子树内”的点对,我们没有计算,
而第二类点对对于根结点是不存在的(都已经是根结点了,不存在不在ta管辖下的点)
我们要考虑的是根结点的儿子,并且给这些点维护第二类贡献
枚举所有子树,先把子树扫一遍清除其在color数组中的所有贡献(change函数)
(因为根结点的颜色一定是“在该点到根这条链上是第一次出现”,所以我们要先
−size[son]
)
void change(int now,int fa,int z)
{
cnt[C[now]]++;
for (int i=st[now];i!=-1;i=way[i].nxt)
{
int v=way[i].y;
if (v!=fa&&!vis[v]) change(v,now,z);
}
if (cnt[C[now]]==1)
{
sum+=(ll)size[now]*z;
color[C[now]]+=(ll)size[now]*z;
}
cnt[C[now]]--;
}
记
tt=size[root]−size[该子树]
这样我们就进入了下一个函数:
dfs_2
void dfs_2(int now,int fa)
{
cnt[C[now]]++;
if (cnt[C[now]]==1)
{
sum-=color[C[now]];
num++;
}
ans[now]+=(ll)sum+num*tt;
for (int i=st[now];i!=-1;i=way[i].nxt)
{
int v=way[i].y;
if (v!=fa&&!vis[v]) dfs_2(v,now);
}
if (cnt[C[now]]==1)
{
sum+=color[C[now]];
num--;
}
cnt[C[now]]--;
}
这里我们就需要利用第一条性质了
我们遍历整棵子树
对于该子树中的每一个点j:
设X=Σcolor[j 到根上(不包括根)的所有颜色]
由于这些颜色在子树中出现过,我们可以通过之后的分治记录ta们的贡献,所以这里就不用计算了
设
num
为j到根上(不包括根)的颜色数
则ans[j]+=sum-X+num*tt
clear
void clear(int now,int fa)
{
cnt[C[now]]=0;
color[C[now]]=0;
for (int i=st[now];i!=-1;i=way[i].nxt)
{
int v=way[i].y;
if (v!=fa&&!vis[v]) clear(v,now);
}
}
清空cnt和color数组,方便下一步的分治