牛客2018多校一 E Removal

最新推荐文章于 2024-08-13 20:00:37 发布

原创最新推荐文章于 2024-08-13 20:00:37 发布 · 307 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#dp

动态规划专栏收录该内容

22 篇文章

订阅专栏

博客围绕从给定n个数的数组中删除m个数，求新数组种类数的问题展开。采用dp方法，初始转移方程会出现重复情况，通过设置数组pre[i]表示该位置数前一次出现的位置，在原有方程基础上减去前一次出现位置前的方案数得到答案。

https://ac.nowcoder.com/acm/contest/139/E
$题意：从给定 n 个数的数组中删除 m 个数，问组成的新数组的种类数$
$题解： n < = 1 e 5, m < = 10 ，想到 d p$
$d p [i] [j] 表示从前 i 个数中删除 j 个数的种类数，$
$很容易想到，如果当前位不删除，那么方案数就是 d p [i - 1] [j - 1] ，如果删除，方案数就是 d p [i - 1] [j]$
$转移方程就是 d p [i] [j] = d p [i - 1] [j] + d p [i - 1] [j - 1]$

$猛敲一发发现样例都过不了，问题就出在这个转移方程会出现重复的情况$
$比如 2312312 ， i = 6 ， j = 3 ，假若删除 i = 4 到 i = 6 ，那么数列变成 2312 ，那么我们发现当 i = 3 ， j = 3 时，我们删除了 i = 1 到 i = 3 ，$ $结果变成了 2312 ，与 i = 6 时的结果重复，问题就在于相同的数字在之前的方案计算中就已经对答案有贡献了$
$这里设置一个数组 p r e [i] 表示 i 这个位置的数 a [i] ，它前一次出现的位置。每次 d p 的时候只需要在原有方程的基础上减去前一次出现的位置前的方案数就是答案。$

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<stdio.h>
#include<string.h>
#include<queue>
#include<cmath>
#include<map>
#include<set>
#include<vector> 
using namespace std;
#define inf 0x3f3f3f3f
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define mem(a,b) memset(a,b,sizeof(a));
#define lowbit(x)  x&-x;  
#define debugint(name,x) printf("%s: %d\n",name,x);
#define debugstring(name,x) printf("%s: %s\n",name,x);
typedef long long ll;
typedef unsigned long long ull;
const double eps = 1e-6;
const int maxn = 1e5+5;
const int mod = 1e9+7;
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,m,k;
ll dp[maxn][15];
int pre[maxn],a[maxn],num[maxn];
int main() {
    while(~scanf("%d%d%d",&n,&m,&k)){
        mem(pre,-1);
        mem(num,0);
        for(int i = 1; i <= n; i++){
            scanf("%d",&a[i]);
            if(pre[i] == -1){
                pre[i] = 0;
            }
            if(!num[a[i]]){
                num[a[i]] = i;
            }else{
                pre[i] = num[a[i]];
                num[a[i]] = i;
            }
        }
        mem(dp,0);
        dp[0][0] = 1;
        for(int i = 1; i <= n; i++){
            dp[i][0] = 1;

            for(int j = 1; j <= m; j++){
                dp[i][j] = (dp[i-1][j-1] + dp[i-1][j] - dp[pre[i]-1][j-(i-pre[i])] + mod)%mod;
            }
        }
        ll ans = 0;
        printf("%lld\n",dp[n][m]%mod);
    }
}