GMOJ 6898. 【2020.11.27提高组模拟】第二题

题目描述

题解

这题考场时想了特别久，花了很多时间，但是只想出了\(O(n^2)\)的做法，满分做法其实不难。

容易发现，一个点如果能变成黑的，当且仅当这个点是黑色或者子树中有两个节点是黑色的。

进一步可以发现，对于这个子树中的点，他们对于这个点的要求是：除自己外的子树有黑色点，或者这个点是黑色。

也就是只要这个点能变成黑色，就能对其子树中所有点产生贡献。

不妨用一个树形dp，将每个点变成黑色的最短时间计算出来。然后按编号从小到大加入点，如果在当前编号下某个点可以变成黑色，就将这个点子树中的点（出这个点以外）对答案的贡献加一，同时将当前最新变色的点对答案的贡献加一。然后就能知道每个点首次加进去的答案了。

但是还需要统计对之前的点的贡献，因为这些点也可能被当前的区间修改，所以再开一个线段树维护一下就行了。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 800010
#define ll long long
#define mo 1000000007
using namespace std;
int n,i,j,bfs[N],fa[N],dfn[N],son[N],num,root,q[N],tot,f[N],g[N],stk[N][2],top;
int x,y;
ll sum,ans;
struct node{
	ll lazy,sum,pk;
}tr[N*5];
struct edge{
	int to,next;
}e[N];
struct pl{
	int time,val;
}dp[N];
int read(){
	int x=0;
	char ch=getchar();
	while (ch<'0'||ch>'9') ch=getchar();
	while (ch>='0'&&ch<='9'){
		x=(x<<1)+(x<<3)+(ch^48);
		ch=getchar();
	}
	return x;
}
void insert_edge(int x,int y){
	tot++;
	e[tot].to=y;
	e[tot].next=q[x];
	q[x]=tot;
}
int cmp(pl x,pl y){
	return x.time<y.time;
}
void update(int x,int l,int r){
	ll p=tr[x].lazy;
	if (l<r){
		int mid=(l+r)/2;
		tr[x*2].sum=(tr[x*2].sum+(mid-l+1)*p)%mo;
		tr[x*2+1].sum=(tr[x*2+1].sum+(r-mid)*p)%mo;
		tr[x*2].lazy=(tr[x*2].lazy+p)%mo;tr[x*2+1].lazy=(tr[x*2+1].lazy+p)%mo;
	}
	tr[x].lazy=0;
}
void change(int x,int l,int r,int l1,int r1){
	if (l1>r1) return;
	update(x,l,r);
	if (l1<=l&&r1>=r){
		tr[x].sum=(tr[x].sum+r-l+1)%mo;
		tr[x].lazy=tr[x].lazy+1;
		return;
	}
	int mid=(l+r)/2;
	if (l1<=mid) change(x*2,l,mid,l1,r1);
	if (r1>mid) change(x*2+1,mid+1,r,l1,r1);
	tr[x].sum=(tr[x*2].sum+tr[x*2+1].sum)%mo;
}
void changepk(int x,int l,int r,int k){
	if (l==r){
		tr[x].pk++;
		return;
	}
	int mid=(l+r)/2;
	if (k<=mid) changepk(x*2,l,mid,k);
	else changepk(x*2+1,mid+1,r,k);
	tr[x].pk=(tr[x*2].pk+tr[x*2+1].pk)%mo;
}
ll gets(int x,int l,int r,int k){
	update(x,l,r);
	if (l==r) return tr[x].sum;
	int mid=(l+r)/2;
	if (k<=mid) return gets(x*2,l,mid,k);
	else return gets(x*2+1,mid+1,r,k);
}
ll getpk(int x,int l,int r,int l1,int r1){
	if (l1>r1) return 0;
	if (l1<=l&&r1>=r) return tr[x].pk;
	int mid=(l+r)/2,pk=0;
	if (l1<=mid) pk=(pk+getpk(x*2,l,mid,l1,r1))%mo;
	if (r1>mid) pk=(pk+getpk(x*2+1,mid+1,r,l1,r1))%mo;
	return pk;
}
int main(){
	freopen("dierti.in","r",stdin);
	freopen("dierti.out","w",stdout);
	n=read();
	for (i=1;i<=n;i++){
		fa[i]=read();
		if (fa[i]==0) root=i;
		else insert_edge(fa[i],i);
	}
	top=1;stk[1][0]=root;stk[1][1]=q[root];
	while (top){
		x=stk[top][0];
		if (stk[top][1]==q[x]){
			f[x]=g[x]=1e9;
			dfn[x]=++num;
		}
		if (!stk[top][1]){
			g[fa[x]]=min(g[fa[x]],min(f[x],x));
			if (g[fa[x]]<f[fa[x]]) swap(f[fa[x]],g[fa[x]]);
			son[x]=num;
			top--;
			continue;
		}
		for (i=stk[top][1];i;i=e[i].next){
			y=e[i].to;
			stk[top][1]=e[i].next;
			stk[++top][0]=y;stk[top][1]=q[y];
			break;
		}
	}
	for (i=1;i<=n;i++) dp[i].time=min(i,g[i]),dp[i].val=i;
	sort(dp+1,dp+n+1,cmp);
	j=0;
	ans=1;
	for (i=1;i<=n;i++){
		change(1,1,n,dfn[i],dfn[i]);
		while (j+1<=n&&dp[j+1].time<=i){
			j++;
			change(1,1,n,dfn[dp[j].val]+1,son[dp[j].val]);
			sum=(sum+getpk(1,1,n,dfn[dp[j].val]+1,son[dp[j].val]))%mo;
		}
		sum=(sum+gets(1,1,n,dfn[i]))%mo;
		changepk(1,1,n,dfn[i]);
		ans=ans*sum%mo;
	}
	printf("%lld\n",ans);
	return 0;
}