noi.ac #528 神树和排列

题目链接：戳我

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
#define MAXN 8010
#define mod 1000000007
#define ll long long
int n,m,a[MAXN],flag[MAXN],cnt[MAXN],f[2][MAXN],sum[MAXN];
int main()
{
	#ifndef ONLINE_JUDGE
	freopen("ce.in","r",stdin);
	#endif
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;++i)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		if(a[x]){puts("0");return 0;}
		if(flag[y]){puts("0");return 0;}
		if(y==n&&x!=n){puts("0");return 0;}
		if(x==n&&y!=n){puts("0");return 0;}
		a[x]=y;flag[y]=1;//a[x]=y表示位置x预定的是y
	}
	//f[i][j]表示统计到前i位，其中最大的是j的方案数
        //根据题目中的排序规则，最大的一定在最右边
	for(int i=1;i<=n;++i) cnt[i]=cnt[i-1]+flag[i];
	f[0][0]=1;
	for(int i=0;i<=n;++i) sum[i]=1;
	for(int i=1;i<=n;++i)
	{
		int now=i&1,pre=now^1;
		memset(f[now],0,sizeof(f[now]));
		//j一定在第i位
		for(int j=i;j<=n;++j)
		{
			if(!a[i-1])//i-1的位置没有被预定
			{
				f[now][j]=((f[now][j]+1ll*f[pre][j]*(j-(i-1)-cnt[j-1])%mod)%mod+sum[j-1])%mod;
				//乘上的系数是该位可以放哪些值
			}
			else if(j>a[i-1])//如果i-1的位置被预定了，且j大于i-1位置预定的值
				f[now][j]=(f[pre][j]+f[pre][a[i-1]])%mod;
		}
		sum[i-1]=0;
		for(int j=i;j<=n;++j) sum[j]=(sum[j-1]+(flag[j]?0:f[now][j]))%mod;
		//sum[j]表示以1...j为最大的值的前缀和
		//如果这个位置被预定了，那么就是0，如果没有，就加上这一位的值
		if(a[i])//如果这个位置被预定了
		{
			flag[a[i]]=0;//a[i]这个数没有预定了，以消除后面统计前缀和的影响
			for(int j=a[i];j<=n;++j) --cnt[j];//cnt[j]表示前j大的数已经被放了几个了,同上
		}
	}
	printf("%d
",f[n&1][n]);
	return 0;
}