定一个字符串 S 和一个字符串 T,计算在 S 的子序列中 T 出现的个数。
一个字符串的一个子序列是指,通过删除一些(也可以不删除)字符且不干扰剩余字符相对位置所组成的新字符串。(例如,“ACE” 是 “ABCDE” 的一个子序列,而 “AEC” 不是)
示例 1:
输入: S = “rabbbit”, T = “rabbit”
输出: 3
解释:
如下图所示, 有 3 种可以从 S 中得到 “rabbit” 的方案。
(上箭头符号 ^ 表示选取的字母)
rabbbit
^^^^ ^^
rabbbit
^^ ^^^^
rabbbit
^^^ ^^^
示例 2:
输入: S = “babgbag”, T = “bag”
输出: 5
解释:
如下图所示, 有 5 种可以从 S 中得到 “bag” 的方案。
(上箭头符号 ^ 表示选取的字母)
babgbag
^^ ^
babgbag
^^ ^
babgbag
^ ^^
babgbag
^ ^^
babgbag
^^^
思路
动态规划,建立数组dp,dp[b][a]表示是s[:b]到t[:a]的方案数量,用b遍历s,用a遍历t
当s[b]!=t[a]时,dp[b][a]=dp[b-1] [a] (只能删掉s[b],等于s[:b-1]到t[:a]的方案数量)
当s[b]==t[a]时,dp[b][a]=dp[b-1][a-1]+dp[b-1][a] (删掉s[b]和不删s[b]的方案数量总和,dp[b-1][a-1] 为不删除s[b]的方案数量,因为s[b]==t[a],将s[:b-1]转换为t[:a-1]即可)
代码
def numDistinct(self, s, t):
dp=[]
for i in range(len(s)+1):dp.append([0]*(len(t)+1))
dp[0][0]=1
for i in range(len(s)+1):dp[i][0]=1
for a in range(1,len(t)+1):
for b in range(a,len(s)+1):
if(s[b-1]==t[a-1]):dp[b][a]=dp[b-1][a-1]+dp[b-1][a]
else:dp[b][a]=dp[b-1][a]
return dp[-1][-1]