【51Nod1296】有限制的排列

计算整数集合{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;
}