来源:力扣(LeetCode)
描述:
给你两个字符串 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 都只包含小写英文字母。
方法一:枚举
思路与算法
题目要求求出字符串 s 与字符串 t 的连续子串中只差一个字符的子串数目,我们枚举 s 与 t 的所有连续子串,然后找其中只含有差一个字符的子串对的数目即可。在实际枚举时,我们可以枚举 s 与 t 的子串的起点 i, j,并依次往后遍历,二者不同的字符个数为 diff,当我们遍历到起点开始的第 k 个字符时:
- 如果 s[i+k] = t[j+k],此时 diff 的数目保持不变;
- 如果 s[i+k] ≠ t[j+k],此时 diff 的数目加 1;
- 如果此时 diff = 0 时,我们继续往后遍历;
- 如果此时 diff = 1 时,此时子串 s[i, ⋯, (i + k)] 与子串 t[j, ⋯, (j + k)] 不同的字符数目为 1,此时计入答案一次;
- 如果此时 diff > 1 时,此时子串 s[i, ⋯, (i + k)] 与子串 t[j, ⋯, (j + k)] 不同的字符数目大于 1,直接退出遍历;
我们最终统计出所有的符合题目要求的子串对即可。
代码:
class Solution {
public:
int countSubstrings(string s, string t) {
int m = s.size(), n = t.size();
int ans = 0;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
int diff = 0;
for (int k = 0; i + k < m && j + k < n; k++) {
diff += s[i + k] == t[j + k] ? 0 : 1;
if (diff > 1) {
break;
} else if (diff == 1) {
ans++;
}
}
}
}
return ans;
}
};
执行用时:0 ms, 在所有 C++ 提交中击败了100.00%的用户
内存消耗:6 MB, 在所有 C++ 提交中击败了83.87%的用户
复杂度分析
时间复杂度:O(m×n×min(m,n)),其中 m,n 分别为字符串 s 与 t 的长度。我们需要枚举 s 与 t 的起点,此时总共有 m×n 对起点,每对起点遍历的长度最多为 min(m, n),因此时间复杂度为 O(m×n×min(m,n))。
空间复杂度:O(1)。
方法二:动态规划
思路与算法
题目要求求出字符串 s 与字符串 t 的连续子串中只差一个字符的子串对的数目,此时我们可以枚举 s 与 t 中不相等的字符对 (s[i], t[j]),并计算以 (s[i], t[j]) 构造的符合题意的子串数目即可。在实际计算时,设以字符 s[i] 与字符 t[j] 为终点且左侧连续相等的最大长度为 dpl[i][j],设以字符 s[i] 与字符 t[j] 为终点且右侧连续相等的最大长度为 dpr[i][j],此时以 (s[i], t[j]) 为字符对且构成只差一个字符的子串对数目为 (dpl[i][j] + 1) × (dpr[i][j] + 1),此时通过枚举所有符合要求的不同的字符对 (s[i], t[j]) 即可计算出所有符合要求的字符串对的数目。
如何计算以 (s[i], t[j]) 为终点左侧和右侧连续相等的最大长度?此时可以参考经典的问题:「最长公共子序列」,通过动态规划实现即可,动态规划递推公式如下:
我们通过枚举所有不同的字符对,并分别计算出以该字符对构成的符合题目要求的字符串数目即可。
代码:
class Solution {
public:
int countSubstrings(string s, string t) {
int m = s.size(), n = t.size();
vector<vector<int>> dpl(m + 1, vector<int>(n + 1));
vector<vector<int>> dpr(m + 1, vector<int>(n + 1));
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
dpl[i + 1][j + 1] = s[i] == t[j] ? (dpl[i][j] + 1) : 0;
}
}
for (int i = m - 1; i >= 0; i--) {
for (int j = n - 1; j >= 0; j--) {
dpr[i][j] = s[i] == t[j] ? (dpr[i + 1][j + 1] + 1) : 0;
}
}
int ans = 0;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (s[i] != t[j]) {
ans += (dpl[i][j] + 1) * (dpr[i + 1][j + 1] + 1);
}
}
}
return ans;
}
};
执行用时:0 ms, 在所有 C++ 提交中击败了100.00%的用户
内存消耗:7.3 MB, 在所有 C++ 提交中击败了33.06%的用户
复杂度分析
时间复杂度:O(m×n),其中 m, n 分别为字符串 s 与 t 的长度。我们计算以两个字符串中的索引 (i, j) 为左终点与右终点最长连续子串长度需要的时间为 O(m×n),枚举两个字符串中不同的字符对,最多有 O(m×n) 个不同的字符对,并计算以该字符对构成的差一个字符的子串数目,每次计算的时间为 O(1),因此总的时间复杂度为 O(m×n)。
空间复杂度:O(m×n)。需要存储两个字符串中的任意索引 (i, j) 为左终点与右终点最长连续子串长度,需要的空间为 O(m×n)。
author:LeetCode-Solution