持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第18天,点击查看活动详情
给定一个字符串 s,计算 s 的 不同非空子序列 的个数。因为结果可能很大,所以返回答案需要对 10^9 + 7 取余 。
字符串的 子序列 是经由原字符串删除一些(也可能不删除)字符但不改变剩余字符相对位置的一个新字符串。
- 例如,
"ace"是 "abcde" 的一个子序列,但"aec"不是。
示例 1:
输入: s = "abc"
输出: 7
解释: 7 个不同的子序列分别是 "a", "b", "c", "ab", "ac", "bc", 以及 "abc"。
示例 2:
输入: s = "aba"
输出: 6
解释: 6 个不同的子序列分别是 "a", "b", "ab", "ba", "aa" 以及 "aba"。
示例 3:
输入: s = "aaa"
输出: 3
解释: 3 个不同的子序列分别是 "a", "aa" 以及 "aaa"。
提示:
1 <= s.length <= 2000s仅由小写英文字母组成
思路
本题可以用动态规划求解。用fn(i)表示s[0]到s[i]不同非空子序列,先考虑一般情况,s[0]到s[i]都是不同字符,如'abc',fn(0) = 1对应的子序列 ['a'], fn(1) = 3对应的子序列['a'、'b'、'ab'], fn(2) = 7对应的子序列['a'、'b'、'ab'、'c'、'ac'、'bc'、'abc'],fn(i-1) = n, 对应的子序列为[s1, s2...sn] ,fn(i) = 2 * n + 1 ,对应的子序列为fn(i-1)对应的子序列、s[i]、fn(i-1)子序列每一项加上s[i],即[s1, s2...sn, s[i], s1s[i], s2s[i]...sns[i]]。
现在再来考虑存在重复的字符,如'abb',fn(0) = 1对应的子序列 ['a'], fn(1) = 3对应的子序列['a'、'b'、'ab'], fn(2) = 5对应的子序列['a'、'b'、'ab'、'bb'、'abb']。用count[j](0<=j<26)表示以a-z结尾字符串的个数,s[k]对应的count[j] = fn(k - 1) + 1,即[s[k], s1s[k], s2s[k]...sns[k]]。现在再来求解fn(i),需要减去重复的字符串,即 fn(i)=2 * fn(i - 1) + 1 - count[k]。
解题
/**
* @param {string} s
* @return {number}
*/
var distinctSubseqII = function (s) {
const MOD = 1000000007;
const counts = new Array(26).fill(0);
const a = "a".charCodeAt(0);
let res = 0;
for (let i = 0; i < s.length; i++) {
let code = s.charCodeAt(i) - a;
let old = res;
res = (res * 2 + 1 - counts[code] + MOD) % MOD;
counts[code] = old + 1
}
return res;
};