牛客网多校1 Removal(dp)

题目:给你一个n长度的序列s,删除m个数后产生的序列有多少种?(n<=1e5,m<=10,si<=10)

dp[i][j]表示i长度,以j数字结尾的方案数

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int maxn=1e5+10;
int s[maxn];
int dp[maxn][11],sum[maxn];///长度为i 结尾为j
int n,m,k;
int main()
{
    ///freopen("in.txt","r",stdin);
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        for(int i=1;i<=n;i++)
            scanf("%d",&s[i]);
        memset(sum,0,sizeof(sum));
        memset(dp,0,sizeof(dp));
        sum[0]=1;
        for(int i=1;i<=n;i++)
        {
            for(int j=i;j>=1&&j>=i-m-1;j--)
            {
                sum[j]+=(sum[j-1]-dp[j][s[i]]+mod)%mod;
                sum[j]%=mod;
                dp[j][s[i]]=sum[j-1];
            }
        }
        int ans=(sum[n-m]+mod)%mod;
        printf("%d\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/dllpxfire/article/details/81131802