51nod 1296 有限制的排列

原题hdu4055

#include <bits/stdc++.h>
using namespace std;

const int MAXN=5050;
const long long mod=1e9+7;
int a[MAXN];
long long dp[2][MAXN];
//dp[i][j]表示i位置填好后，末尾为j的答案 

int main()
{
	long long n,k,l,x,flag,i,j,ans,sum;
	while(~scanf("%lld%lld%lld",&n,&k,&l))
	{
		memset(a,0,sizeof(a));
		memset(dp,0,sizeof(dp));
		flag=1;
		for(i=1;i<=k;i++)
		{
			scanf("%lld",&x);
			a[x]--;
		}
		for(i=1;i<=l;i++)
		{
			scanf("%lld",&x);
			if(a[x]<0)	flag=0;
			a[x]++;
		}
		if(!flag)
		{
			puts("0");
			continue;
		}
		dp[0][1]=1;
		for(i=1;i<n;i++)
		{
			if(a[i]&&a[i]==a[i-1])
			{
				for(j=1;j<=i+1;j++)
					dp[i&1][j]=0;
			} 
			else if(a[i]==1||a[i-1]==-1)
			{
				dp[i&1][1]=0;
				for(j=2;j<=i+1;j++)
				{
					dp[i&1][j]=(dp[i&1][j-1]+dp[i&1^1][j-1])%mod;
				}
			}
			else if(a[i]==-1||a[i-1]==1)
			{
				for(j=i;j>=1;j--)
					dp[i&1][j]=(dp[i&1][j+1]+dp[i&1^1][j])%mod;
			}
			else
			{
				sum=0;
				for(j=1;j<=i;j++)
					sum=(sum+dp[i&1^1][j])%mod;
				for(j=1;j<=i+1;j++)
					dp[i&1][j]=sum;
			}
		} 
		ans=0;
		for(i=1;i<=n;i++)
			ans=(ans+dp[n&1^1][i])%mod;
		printf("%lld\n",ans);
	}
}