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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值