removal—牛客暑假多校1——dp

链接：https://www.nowcoder.com/acm/contest/139/E
来源：牛客网
 

题目描述

Bobo has a sequence of integers s1, s2, ..., sn where 1 ≤ si ≤ k.
Find out the number of distinct sequences modulo (109+7) after removing exactly m elements.

输入描述:

The input consists of several test cases and is terminated by end-of-file.
The first line of each test case contains three integers n, m and k.
The second line contains n integers s1, s2, ..., sn.

输出描述:

For each test case, print an integer which denotes the result.

 

示例1

输入

复制

3 2 2
1 2 1
4 2 2
1 2 1 2

输出

复制

2
4

备注:

* 1 ≤ n ≤ 105
* 1 ≤ m ≤ min{n - 1, 10}
* 1 ≤ k ≤ 10
* 1 ≤ si ≤ k
* The sum of n does not exceed 106.

题意就是给定一段长为n的序列。去掉m个数之后，有几种不同的序列。

思路参考：

https://blog.csdn.net/litmxs/article/details/81141606

很容易想到，dp[i][j]表示到i位置时，去掉了j个数字时的不同序列的个数。

那么dp[ i ] [ j ]=dp【i-1】【j-1】+dp【i-1】【j】个。

很明显，当序列为(....1,2,3,4,1)的时候，删掉（.....1，2，3，4，1......）和（....1，2，3，4，1 .......）的这种情况是重复的。

那么就是说，我们要将之前算过的那个第一个1出现的那块去掉。

假设删除5个，那么在i枚举到最后一个1的位置时，肯定算的时候有一种情况包含 “2，3，4，1 ”和“1，2，3，4”我就想剪掉中间的一种。于是就可以想到，dp【i-pre【i】-1】【j-d】（d=i-pre【i】）就是前边选j-d个，后边选d个。

pre【i】指的是a【i】在位置i之前出现的距离i最近的位置

此处还需要细细琢磨。

 

具体代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 1e5+100, mod = 1e9+7;
int n, m, k, s[maxn];
long long dp[maxn][20];
int pre[maxn];
int prex[20];
int main()
{
    while(scanf("%d%d%d", &n, &m, &k) == 3)
    {
        memset(dp, 0, sizeof(int)*20*(n+1));
        memset(prex, 0, sizeof(prex));
        for(int i=1; i<=n; ++i)
        {
            scanf("%d", &s[i]);
            pre[i] = prex[s[i]];
            prex[s[i]] = i;
        }
        dp[0][0] = 1;
        for(int i=1; i<=n; ++i)
        {
            dp[i][0] = 1;
            int d = i - pre[i];
            for(int j=1; j<=m&&j<=i; ++j)
            {
                dp[i][j] = dp[i-1][j] + dp[i-1][j-1];
                if(pre[i] && d<=j)
                    dp[i][j] -= dp[pre[i]-1][j-d];
                
                ///dp[i][j] %= mod;
                dp[i][j] += mod;
                dp[i][j] %= mod;
                
            }
        }
        printf("%lld\n", dp[n][m]);
    }

    return 0;
}

如果dp[i][j]是int的话，那么就要先mod再加。就是将我的注释部分加上。因为会超longlong