【51Nod1202】子序列个数(简单dp)

最新推荐文章于 2023-03-16 11:21:08 发布

原创最新推荐文章于 2023-03-16 11:21:08 发布 · 1.3k 阅读

3 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#dp

dp 专栏收录该内容

36 篇文章

订阅专栏

博客讨论了一种使用动态规划求解给定序列所有子序列个数的方法。通过分析2^n子序列的来源，博主提出一个时间复杂度为O(n)的解决方案，利用动态规划数组dp[i]表示前i项子序列的个数，并通过维护序列中每个元素的最新出现位置来处理重复子序列。最终，答案是dp[n] - 1，对10^9 + 7取余。

记录一个菜逼的成长。。

题目链接
给定一个正整数序列，序列中元素的个数和元素值大小都不超过105, 求其所有子序列的个数。注意相同的只算一次：例如 {1,2,1}有子序列{1} {2} {1,2} {2,1}和{1,2,1}。最后结果对10^9 + 7取余数。

分析：万能枚举，时间复杂度O(2^n)——因为每一项都可以选择取或者不取嘛。这个不可行，n太大了。

我们想想动态规划？严格来讲这个题不是一个dp的问题，因为它是一个数数（组合数学）的问题，不是一个优化的问题。我们来想想2^n个子序列是怎么来的——之前已经有2^(n-1)个了，再加上最后一个或者不加上最后一个，还有两种选择。

假设dp[i]表示前i项形成的子序列（含空）的个数。下标从1开始，初值是dp[0] = 1，对应代表空子序列。我们考虑第i项，如果所有的数都不相等，应该有dp[i] = dp[i – 1] * 2，其实就是考虑把第i个数放到最后或者不放到最后的情况。
然而，因为有可能有相同的数，我们假设第i个数出现之前最近的在j (j < i)位置也出现过，那么实际上我们这种简单＊2会有重复，哪些子序列重复呢？我们设数列为a, 并且下标从1开始。

原来恰好以第j个数结尾的那些被我们算了两次，因为以第j个数结尾可以换成以第i个数结尾是一样的。如何计算出这个数的个数呢？其实这个数等于dp[j – 1]，因为前面(j – 1)个数的子序列最后跟上第j个数就可以了。

于是我们有了递推式：

dp[i] = dp[i – 1] * 2 如果a[i]不在之前出现
dp[i] = dp[i – 1] * 2 – dp[j – 1]，如果a[i]最近在j的位置出现过。
这里写图片描述
这里有一个问题：我们如何找到a[i]最近在哪里出现呢？我们可以用一个hash，更简单的方式是直接把a[i]放到数组下标里，记录它最后一次出现的位置，因为数据范围并不大，可以承受。所以我们有了一个时间复杂度O(n)的动态规划解法。

注意最后答案是dp[n] – 1 因为我们需要减掉一个全空子序列。

伪代码：

输入: a[1..n]
辅助变量： have[x]表示目前x在a中最后一次出现的位置，如果没出现就是0，否则是一个正数。

dp[0] = 1
for  i = 1 to n do
    dp[i] = dp[i – 1] * 2
    if have[a[i]] > 0 then dp[i] -= dp[have[a[i]] – 1]
return dp[n] – 1

另外，题目为了避免大数运算采用了取余数操作，请别忘记取余数操作。

//#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <list>
#include <deque>
#include <cctype>
#include <bitset>
#include <cmath>
#include <cassert>
using namespace std;
#define ALL(v) (v).begin(),(v).end()
#define cl(a,b) memset(a,b,sizeof(a))
#define bp __builtin_popcount
#define pb push_back
#define mp make_pair
#define fin freopen("D://in.txt","r",stdin)
#define fout freopen("D://out.txt","w",stdout)
#define lson t<<1,l,mid
#define rson t<<1|1,mid+1,r
#define seglen(t) (node[t].r-node[t].l+1)
#define pi 3.1415926
#define exp  2.718281828459
#define lowbit(x) (x)&(-x)
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;
typedef pair<LL,LL> PLL;
typedef vector<PII> VPII;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
template <typename T>
inline void read(T &x){
    T ans=0;
    char last=' ',ch=getchar();
    while(ch<'0' || ch>'9')last=ch,ch=getchar();
    while(ch>='0' && ch<='9')ans=ans*10+ch-'0',ch=getchar();
    if(last=='-')ans=-ans;
    x = ans;
}
const int maxn = 100000 + 10;
LL a[maxn],index[maxn],dp[maxn];
int main()
{
    int n;
    while(~scanf("%d",&n)){
        cl(index,0);
        for( int i = 1; i <= n; i++ ){
            scanf("%lld",a+i);
        }
        dp[0] = 1;
        for( int i = 1; i <= n; i++ ){
            dp[i] = dp[i-1] * 2;
            dp[i] %= MOD;
            if(index[a[i]] > 0)dp[i] -= dp[index[a[i]]-1];
            index[a[i]] = i;
            dp[i] = (dp[i] + MOD) % MOD;
        }
        printf("%lld\n",dp[n] - 1);
    }
    return 0;
}