Newcoder 26 B.点权和（机智）

Description

给你一棵树，最开始点权为$0$，每次将与一个点$x$树上距离$\le 1$的所有点点权$+1$，之后询问这些点修改后的点权和.

Input

第一行两个数$n$和$m$

第二行$n-1$个数，第$i$个数$fa[i+1]$表示$i+1$点的父亲编号，保证$fa[i+1]<i+1$

第三行$m$个数，每个数$x$次表示这次操作的点是$x$

$(n\le 10^5,m\le 10^6)$

Output

输出一个数，即这$m$次操作的答案的$hash$值

如果是第$i$次操作，这次操作结果为$ans$,则这个$hash$值加上$i\cdot ans$

输出$hash$值对$19260817$取模的结果

Sample Input

6 3
1 1 2 3 3
1 2 3

Sample Output

34

Solution

查询一点$u$的答案分为以下

1.由于$u$被操作$cnt[u]$次使得$u$以及$u$的所有邻接点答案加一，该部分贡献为$cnt[u]\cdot (du[u]+1)$

2.由于$u$的父亲被操作$cnt[fa[u]]$次，使得$u$和$fa[u]$答案都加一，该部分贡献为$2\cdot cnt[fa[u]]$

3.由于$u$的爷爷被操作$cnt[fa[fa[u]]]$次，使得$fa[u]$答案加一，该部分贡献为$cnt[fa[fa[u]]]$

4.由于$u$的兄弟被操作$scnt[fa[u]]-cnt[u]$次，使得$fa[u]$答案加上该部分，该部分贡献$scnt[fa[u]]-cnt[u]$，其中$scnt[fa[u]]$表示$fa[u]$的儿子们被操作的次数之和

5.由于$u$的儿子被操作使得$u$的答案加上对应值，此部分可以在其儿子被操作时直接累加到$u$的答案上

对$u$点操作一次的影响为

1.对$u$父亲的影响，使得$fa[u]$的答案加二

2.对$u$爷爷的影响，使得$fa[fa[u]]$的答案加一

故只需要维护$cnt[u],scnt[u]$以及统计答案的$ans[u]$即可，时间复杂度$O(n+m)$

Code

#include<cstdio>
using namespace std;
typedef long long ll;
const int maxn=100005;
#define mod 19260817
int add(int x,int y)
{
	x+=y;
	if(x>=mod)x-=mod;
	return x;
}
int mul(int x,int y)
{
	ll z=1ll*x*y;
	return z-z/mod*mod;
}
int n,m,f[maxn],du[maxn],cnt[maxn],scnt[maxn],ans[maxn];
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=2;i<=n;i++)scanf("%d",&f[i]),du[i]++,du[f[i]]++;
	int sum=0;
	for(int i=1;i<=m;i++)
	{
		int u,res=0;
		scanf("%d",&u);
		ans[u]=add(ans[u],du[u]+1);
		if(f[u])ans[f[u]]=add(ans[f[u]],2),scnt[f[u]]=add(scnt[f[u]],1);
		if(f[f[u]])ans[f[f[u]]]=add(ans[f[f[u]]],1);
		cnt[u]=add(cnt[u],1);
		if(f[u])
		{
			res=add(res,mul(2,cnt[f[u]]));
			res=add(res,add(scnt[f[u]],mod-cnt[u]));
		}
		if(f[f[u]])res=add(res,cnt[f[f[u]]]);
		res=add(res,ans[u]);
		sum=add(sum,mul(i,res)); 
	}
	printf("%d\n",sum);
	return 0;
}