【JZOJ 省选模拟】6655.三国学者

题目

Description

Input

Output

Sample Input
5 5 3
1 2 3 3
3 2 2 1 2
3 2 2 2 2

Sample Output
12

Data Constraint
对于所有数据，有Ai<L<=N<=10^6, M<=2*10^6, fai<i,vi<L.

思路

先考虑暴力dp
设 f[i][j]为从根节点到i的路径上有多少条结尾权值为j的路径，最后加上所有wx=1的节点的f[x][2]就行。

考虑优化
我们发现每到⼀个节点，f[i] 相对于 f[f[ai]]只改变了一个位置。那么我们只需要通过dfs来维护一个全局的dp数组，来表示出每个节点的f即可。
即在进入个点时修改f ，退出时把这个点的影响消除。

现在的问题是如何快速地计算gx
看起来不能用桶直接从子树搞上来，因为涉及到合并的问题。
但是大佬又提供了一种神奇的思路：直接将g值加入桶中，从x往下递归之前先记一下，递归回来之后作差，就可以得到子树对gx的贡献。

代码

#include<bits/stdc++.h>
#define N 1000005
#define ll long long 
#define mod 1000000007
using namespace std;

int n,m,L,lim,fa[N],a[N],b[N],now;
int cnt,e[N],nx[N],ls[N];
ll sum,vf[N],vg[N],Ans,d[N];

void dfs(int x)
{
	if(x<=lim) (vf[b[x]]+=vf[b[x]+1])%=mod; 
	ll g0=vg[a[x]-1],g1=vg[b[x]-1];
	for(int i=ls[x]; i; i=nx[i]) dfs(e[i]);
	g0=vg[a[x]-1]-g0,g1=vg[b[x]-1]-g1;
	if(a[x]==1) g0=1; if(b[x]==1) g1=1;
	if(x<=lim) {
		(vf[b[x]]-=vf[b[x]+1])%=mod;
		d[x]=g1%mod*vf[b[x]+1]-g0%mod*vf[a[x]+1];
	}
	(vg[a[x]]+=g0)%=mod;
}

int main(){
	freopen("scholar.in","r",stdin);
	freopen("scholar.out","w",stdout);
	scanf("%d%d%d",&n,&m,&L);
	for(int i=2; i<=n; i++) scanf("%d",&fa[i]);
	for(int i=n; i>=2; i--) cnt++,e[cnt]=i,nx[cnt]=ls[fa[i]],ls[fa[i]]=cnt;
	for(int i=1; i<=n; i++) scanf("%d",&a[i]);
	
	now=0; int tp=0; vf[L+1]=1;
	while(m)
	{
		lim=min(n,m); m-=lim;
		for(int i=1; i<=lim; i++) scanf("%d",&b[i]);
		if(tp) for(int i=1; i<=n; i++) vg[a[i]]=0; else tp=1;
		dfs(1);
		sum=vg[L];
		for(int i=1; i<=lim; i++)
		{
			(sum+=d[i])%=mod,a[i]=b[i];
			Ans+=sum*(now+i)%mod;
		}
		now+=lim;
	}
	printf("%lld",(Ans%mod+mod)%mod);
}