给定一个字符串 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];
}
}