子序列的个数

题目详情

本题同样来自caopengcs，只要你有兴趣，每个人都可以出题（出题入口在主页右侧边栏“贡献题目”->“我要发布”内），以下是题目详情：

子序列的定义：对于一个序列a=a[1],a[2],......a[n]，则非空序列a'=a[p1],a[p2]......a[pm]为a的一个子序列，其中1<=p1<p2<.....<pm<=n。

例如：4,14,2,3和14,1,2,3都为4,13,14,1,2,3的子序列。

对于给出序列a，有些子序列可能是相同的，这里只算做1个，要求输出a的不同子序列的数量。

输入： 长度为n的数组1<=n<=100,数组元素0<=a[i]<=110

输出：子序列 的个数对1000000007取余数的结果（由于答案比较大，输出Mod 1000000007的结果即可）。

函数头部：

C/C++:

  int run(cons int *a,int n);

java

  public static int run(int [] a);

这个题把我难住了,虽然我知道要用动态规划,但是一直没想到怎么用,

琢磨出来一个

dp[i][j][k]//在1到i的区段内，选取j个元素的，以k为结尾元素的自序列的个数。

dp[i + 1][j + 1][a[i]] += dp[i][j][k];//在dp[i][j][k]的尾巴上加a[i]

dp[i + 1][j][k] += dp[i][j][k](k != a[i])//放弃a[i]，当k == a[i]时，因为会有重复，故不算

拿[1， 2， 3， 4]，[1，2， 3， 3， 2]测试也都是对的，

信心满满地提交了，没过，有点伤心。在网上找了答案，发现思路好简单，其实我在自己推的时候也差点发现了2*x-1的规律，不过自己没有细想。

现在琢磨一下，感觉自己的方程在复杂情况下说不太清楚。但还是不知道为什么会出错。。。

以下是出错了的代码和正确的代码

#include <cstdio>
#include <cstring>
using namespace std;

#define MOD 1000000007
#define N 120

int dp[N][N][N];


int run(const int *a, int n)
{
    int sum = 0;
    int i, j, k;
    memset(dp, 0, sizeof(dp));
    dp[1][1][a[0]] = 1;
    for(i = 1; i<= n; i++)//在dp上设初值,在第i - 1位并丢弃前i - 2位的值为1 
    {
        dp[i][1][a[i - 1]] = 1;
    }
    for(i = 1; i < n; i++)//从dp[i]推dp[i + 1]
    {
        for(j = 1; j <= i; j++)
        {
            for(k = 0; k < N; k++)
            {
                if(k != a[i])
                {
                    dp[i + 1][j][k] = (dp[i + 1][j][k] + dp[i][j][k]) % MOD;
                }
                dp[i + 1][j + 1][a[i]] = (dp[i + 1][j + 1][a[i]] + dp[i][j][k]) % MOD;
            }
        }
    }
    for(j = 1; j <= n; j++)
    {
        for(k = 0; k <= N; k++)
        {
            sum = (sum + dp[n][j][k]) % MOD;
        }
    }
    return sum;
}

//start 提示：自动阅卷起始唯一标识，请勿删除或增加。
int main()
{   
    int a[] = {1, 3, 4, 4, 3};
    printf("%d\n", run(a, sizeof(a) / sizeof(a[0])));
    return 0;
} 
//end //提示：自动阅卷结束唯一标识，请勿删除或增加。   

//正确的代码

//dp[i] = dp[i - 1] * 2 + 1 (a[i - 1] 第一次出现)

//dp[i] = dp[i - 1] * 2 - (dp[a[i - 1]上一次出现 - 1])

#include<cstdio>
#include<cstring>

#define N 120
#define MOD 1000000007

int run(const int *a, int n)
{
    int i;
    long long subArray[N] = {0};
    int lastSameIndex[N];
    memset(lastSameIndex, 0, sizeof(lastSameIndex));
    for(i = 1; i <= n; i++)
    {
        if(lastSameIndex[a[i - 1]] == 0)
        {
            subArray[i] = subArray[i - 1] * 2 + 1;
        }
        else
        {
            subArray[i] = subArray[i - 1] * 2 - (subArray[lastSameIndex[a[i - 1]] - 1]) + MOD;
        }
        subArray[i] %= MOD;
        lastSameIndex[a[i - 1]] = i;
    }
    return subArray[n];
}

//start 提示：自动阅卷起始唯一标识，请勿删除或增加。
int main()
{   
    int a[] = {1, 3, 4, 4, 3};
    printf("%d\n", run(a, sizeof(a) / sizeof(a[0])));
    return 0;
} 
//end //提示：自动阅卷结束唯一标识，请勿删除或增加。