fzu 2129 子序列个数 （dp）

 Problem 2129 子序列个数

Accept: 25    Submit: 63
Time Limit: 2000 mSec    Memory Limit : 32768 KB

 Problem Description

子序列的定义：对于一个序列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，请输出不同的子序列的个数。(由于答案比较大，请将答案mod 1000000007)

 Input

输入包含多组数据。每组数据第一行为一个整数n(1<=n<=1,000,000)，表示序列元素的个数。

第二行包含n个整数a[i] (0<=a[i]<=1,000,000)表示序列中每个元素。

 Output

输出一个整数占一行，为所求的不同子序列的个数。由于答案比较大，请将答案mod 1000000007。

 Sample Input

41 2 3 2

 Sample Output

13

 Hint

其中40%数据点1<=n<=1000。

 Source

福州大学第十届程序设计竞赛

思路：

设ans[i]为以前i个数为结尾的序列的总数，last[x]为以数x为结尾，长度大于1的序列的总数，vis[x]表示数x是否出现过,第i个数a[i]=x。


若x没有出现过，则以第i个数为结尾的序列的个数为 ans[i-1]+1  (以 前i个数为结尾的序列后面加上一个x,在加上一个长度为1的序列x)
若x出现过，则以第i个数为结尾的序列的个数为  ans[i-1]-last[x]  (以 前i个数为结尾的序列后面加上一个x，但因为x曾经出现过，所以减去重复的部分last[x]，另外取模的时候要注意加上mod。


然后更新last[x]的值，显然这个值为 ans[i-1].


最后ans[n]即为所求的答案。

#include <cstdio>
#include <cstring>
#include <algorithm>
const long M=1100000;
const long mod=1000000007;
long last[M],vis[M],n,ans;
int main(){
	while (~scanf("%d",&n)){
		memset(vis,0,sizeof(vis));
		memset(last,0,sizeof(last));
		ans=0;
		for (long i=1;i<=n;++i){
			long x; 
			scanf("%d",&x);
			if (!vis[x]){
				last[x]=ans;
				ans=(ans+ans+1)%mod;
				vis[x]=1;
			}
			else{
				long temp=ans;
				ans=((temp-last[x]+mod)%mod+ans)%mod;
				last[x]=temp;
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}