JAVA程序设计：不同的子序列 II（LeetCode：940）

给定一个字符串 S，计算 S 的不同非空子序列的个数。

因为结果可能很大，所以返回答案模 10^9 + 7.

 

示例 1：

输入："abc"
输出：7
解释：7 个不同的子序列分别是 "a", "b", "c", "ab", "ac", "bc", 以及 "abc"。
示例 2：

输入："aba"
输出：6
解释：6 个不同的子序列分别是 "a", "b", "ab", "ba", "aa" 以及 "aba"。
示例 3：

输入："aaa"
输出：3
解释：3 个不同的子序列分别是 "a", "aa" 以及 "aaa"。
 

 

提示：

S 只包含小写字母。
1 <= S.length <= 2000

思路：我们令dp[i]表示前i个字符能组成的不同子序列，为了方便起见，我们先考虑可以存在空序列并且序列不存在重复字母的情况，此时很容易想到转移方程：dp[i]=dp[i-1]*2，因为每个字符只有选和不选两种情况嘛。

但是本题不要求有空序列，并且原字符串是有重复字符的，那我们应该怎么办呢？我们考虑一个样例：abab。若此时i=2，即前边已组成了" ","a","ab","b","ba","aba",此时需要添加b，可以发现b一定不会添加到“a”之后，也不会添加到“ ”之后，因为这会导致重复的现象，而导致重复的数量刚好是dp[0],此时我们就可以得到最终的转移方程了：

dp[i]=dp[i-1]*2-dp[last[S[k]]-1]

其中last[s[i]]表示s[i]上次出现的位置。又因为题目要求不能包含空字符串，因此最后所求的答案再减一即可。

class Solution {
    public int distinctSubseqII(String S) {
    	
    	int n=S.length();
    	int mod=1000000007;
    	int[] dp=new int[n+1];
    	int[] last=new int[26];
    	
    	dp[0]=1;
    	Arrays.fill(last, -1);
    	
    	for(int i=0;i<n;i++) {
    		int x=S.charAt(i)-'a';
    		dp[i+1]=dp[i]*2%mod;
    		if(last[x]>=0)
    			dp[i+1]-=dp[last[x]];
    		dp[i+1]%=mod;
    		last[x]=i;
    	}
    	
    	dp[n]--;
    	
    	if(dp[n]<0) dp[n]+=mod;
    	
    	return dp[n];
    }
}