统计只差一个字符的子串数目

题目描述

给你两个字符串 s 和 t ，请你找出 s 中的非空子串的数目，这些子串满足替换 一个不同字符 以后，是 t 串的子串。换言之，请你找到 s 和 t 串中 恰好 只有一个字符不同的子字符串对的数目。

比方说， “computer” and “computation” 只有一个字符不同： ‘e’/‘a’ ，所以这一对子字符串会给答案加 1 。

请你返回满足上述条件的不同子字符串对数目。

一个 子字符串 是一个字符串中连续的字符。

示例 1：

输入：s = “aba”, t = “baba”
输出：6
解释：以下为只相差 1 个字符的 s 和 t 串的子字符串对：
(“aba”, “baba”)
(“aba”, “baba”)
(“aba”, “baba”)
(“aba”, “baba”)
(“aba”, “baba”)
(“aba”, “baba”)
加粗部分分别表示 s 和 t 串选出来的子字符串。
示例 2：
输入：s = “ab”, t = “bb”
输出：3
解释：以下为只相差 1 个字符的 s 和 t 串的子字符串对：
(“ab”, “bb”)
(“ab”, “bb”)
(“ab”, “bb”)
加粗部分分别表示 s 和 t 串选出来的子字符串。
示例 3：
输入：s = “a”, t = “a”
输出：0
示例 4：

输入：s = “abe”, t = “bbc”
输出：10

提示：

1 <= s.length, t.length <= 100
s 和 t 都只包含小写英文字母。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/count-substrings-that-differ-by-one-character
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

动态规划。
定义dp[s.length()+1][t.length()+1][2]
状态： dp[i][j][k]表示以字符s[i-1]结尾，t[j-1]结尾的s和t的子字符串中相差k个字符的子串对数目
在本题中k=0或1
初始值：
dp[0][j][0]=0
dp[0][j][1]=0
dp[i][0][0]=0
dp[i][0][1]=0
状态转移方程：
如果s[i]=t[j]
dp[i+1][j+1][0]=dp[i][j][0]+1
dp[i+1][j+1][1]=dp[i][j][1]
如果s[i]!=t[j]
dp[i+1][j+1][0]=0
dp[i+1][j+1][1]=dp[i][j][0]+1

统计所有的dp[i][j][1]之和为返回值

代码

class Solution {
	public int countSubstrings(String s, String t) {
		int sl=s.length();
		int tl=t.length();
		int[][][] dp = new int[sl+1][tl+1][2];
		for(int i=0;i<sl;i++){
			dp[i][0][0]=0;
			dp[i][0][1]=0;
		}
		for(int j=0;j<tl;j++){
			dp[0][j][0]=0;
			dp[0][j][1]=0;
		}
		int ans=0;
		for(int i=0;i<sl;i++){
			for(int j=0;j<tl;j++){
				if(s.charAt(i)==t.charAt(j)){
					dp[i+1][j+1][0]=dp[i][j][0]+1;
					dp[i+1][j+1][1]=dp[i][j][1];
				}else{
					dp[i+1][j+1][0]=0;
					dp[i+1][j+1][1]=dp[i][j][0]+1;
				}
				ans+=dp[i+1][j+1][1];
			}
		}
		return ans;

	}
}