计算整数集合{1,2,3,4, …. N }满足下列条件的的排列个数:
在位置a1, a2, …, aK小于其邻居(编号从0开始)。
在位置b1, b2, …, bL大于其邻居。
输出符合条件的排列数量Mod 1000000007的结果。例如:N = 4,a = {1}, b = {2},符合条件的排列为:
2 1 4 3
3 2 4 1
4 2 3 1
3 1 4 2
4 1 3 2
Input
第1行:3个数N, K, L,分别表示数组的长度,限制a的长度,限制b的长度(1 <= N <= 5000, 1 <= K, L <= N)。
第2 - K + 1行:每行一个数,对应限制a的位置(1 <= ai <= N - 2)
第K + 2 - K + L + 1行:每行一个数,对应限制b的位置(1 <= bi <= N - 2)
Output
输出符合条件的排列数量Mod 1000000007的结果。
Input示例
4 1 1
1
2
Output示例
5
题解
dp[i][j]表示i位置,插入j的方案数。我们可以记录每个位置与前一个的位置关系进行转移。
1,i比i-1位置大,dp[i][j]+=dp[i-1][1~j-1]
2,i比i-1位置小,dp[i][j]+=dp[i-1][j~i-1](此处是一个插入技巧:第i个位置插入1~i,当插入的数为j时,把j~i-1的数加1,这样就不会重复插入同一个数了)
3,dp[i][j]+=dp[i-1][1~i-1]
可用前缀和优化至N^2
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<set>
#include<ctime>
#include<vector>
#include<cmath>
#include<algorithm>
#include<map>
#include<queue>
#define mod 1000000007
#define ll long long
#define N 10000005
#define inf 0x7fffffff
using namespace std;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int n,k,l,dp[5005],sum[5005],flag[5005];
int main()
{
n=read();k=read();l=read();
for (int i=1;i<=k;i++)
{
int x=read()+1;
flag[x]=1;flag[x+1]=2;
}
for (int i=1;i<=l;i++)
{
int x=read()+1;
flag[x]=2;flag[x+1]=1;
}
sum[1]=dp[1]=1;
for (int i=2;i<=n;i++)
{
for (int j=1;j<=i;j++)
{
if (flag[i]==1) dp[j]=(sum[i-1]-sum[j-1]+mod)%mod;
else if (flag[i]==2) dp[j]=sum[j-1];
else dp[j]=sum[i-1];
}
for (int j=1;j<=i;j++)
sum[j]=(sum[j-1]+dp[j])%mod;
}
printf("%d",sum[n]);
return 0;
}