nowcoder NC17137 - Removal（dp）

最新推荐文章于 2024-10-09 22:28:58 发布

Altria Vin

最新推荐文章于 2024-10-09 22:28:58 发布

阅读量150

点赞数

分类专栏： dp 文章标签：算法

本文链接：https://blog.csdn.net/qq_43423395/article/details/105785517

版权

dp 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

题目链接：https://ac.nowcoder.com/acm/problem/17137

题意：给你 $n$ 个数( $1\le a_i \le k$ )，问你删掉 $m$ 个数以后得到的不同的子序列的个数对 $1 e 9 + 7$ 取模的结果。

思路：首先我们可以考虑一个字问题：

给你 $n$ 个数( $1\le a_i \le k$ )，问你删掉 $m$ 个数以后得到的子序列的个数（包括重复的子序列）对 $1 e 9 + 7$ 取模的结果。

拿着就是一道很简单的 $d p$ 问题啦，我们设 $d p [i] [j]$ 表示前 $i$ 个数里删掉 $j$ 个数得到的子序列的个数，那么就有
$d p [i] [j] = d p [i - 1] [j] + d p [i - 1] [j - 1]$
之后我们在考虑如何删掉重复的子序列。
我们可以用一个数组 $p r e []$ 去记录第 $i$ 个数他前面的第一个相同的数所在的位置。
然后我们在考虑一个子问题：

就是在不考虑长度的情况下子序列重合的个数。

那很明显就是 $d p [p r e [i] - 1]$
那现在考虑删掉个数的情况呢？
假设我们现在已经删掉了 $j$ 个数，那么 $d p [p r e [i] - 1] [j - (i - p r e [i])]$ 是不是就是重合的方案数呢？因为我们要把 $i - p r e [i]$ 之间的数全部删掉才行呀！！！

比如

2 3 4 1 5 6 1

对于这组样例，重复的序列是

2 3 4 1

就是只有把两个1之间的5 6删掉才有重复的序列。
那这一题就结束啦。

AC代码如下：

#include <set>
#include <map>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

#define LL long long
#define pii pair<int,int>
#define sd(x) scanf("%d",&x)
#define slld(x) scanf("%lld",&x)
#define pd(x) printf("%d\n",x)
#define plld(x) printf("%lld\n",x)
#define rep(i,a,b) for(int i = (a) ; i <= (b) ; i++)
#define per(i,a,b) for(int i = (a) ; i >= (b) ; i--)
#define mem(a) memset(a,0,sizeof(a))
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
#define fast_io ios::sync_with_stdio(false)

const int INF = 1e9 + 10;
const LL mod = 1e9 + 7;
const int maxn = 1e5 + 7;

inline int read() {
    char c = getchar();
    int x = 0, f = 1;
    while (c < '0' || c > '9') {
        if (c == '-') f = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9') {
        x = x * 10 + (c - '0');
        c = getchar();
    }
    return x * f;
}

int head[maxn << 1];
struct Edge {
    int to, next;
}edge[maxn << 1];
int tot;

void init() {
    memset(head, -1, sizeof(head));
    tot = 0;
}

void addedge(int u, int v) {
    edge[tot].to = v;
    edge[tot].next = head[u];
    head[u] = tot++;
}

LL qpow(LL a, LL b, LL p) {
    LL res = 1;
    while (b) {
        if (b & 1) res = (res * a) % p;
        a = (a * a) % p;
        b >>= 1;
    }
    return res;
}

LL dp[maxn][20];
LL pre[maxn];
LL c[maxn];
LL a[maxn];

int main() {
    int n,m,k;
    while(~scanf("%d%d%d",&n,&m,&k)) {
        mem(pre), mem(c);
        rep(i,1,n) {
            slld(a[i]);
            pre[i] = c[a[i]];
            c[a[i]] = i;
        }
        rep(i,0,n) dp[i][i] = dp[i][0] = 1;

        rep(i,1,n) {
            rep(j,1,m) {
                dp[i][j] = (dp[i-1][j] + dp[i-1][j-1]) % mod;
                if(pre[i] > 0 && j >= i - pre[i]) {
                    dp[i][j] = (dp[i][j] - dp[pre[i]-1][j-(i-pre[i])] + mod) % mod;
                }
            }
        }
        plld(dp[n][m]);
    }
    return 0;
}