【LetMeFly】1358.包含所有三种字符的子字符串数目:滑动窗口(两种写法直接推荐方法二)
力扣题目链接:https://leetcode.cn/problems/number-of-substrings-containing-all-three-characters/
给你一个字符串 s ,它只包含三种字符 a, b 和 c 。
请你返回 a,b 和 c 都 至少出现过一次的子字符串数目。
示例 1:
输入:s = "abcabc"
输出:10
解释:包含 a,b 和 c 各至少一次的子字符串为 "abc", "abca", "abcab", "abcabc", "bca", "bcab", "bcabc", "cab", "cabc" 和 "abc" (相同字符串算多次)。
示例 2:
输入:s = "aaacb"
输出:3
解释:包含 a,b 和 c 各至少一次的子字符串为 "aaacb", "aacb" 和 "acb" 。
示例 3:
输入:s = "abc"
输出:1
提示:
3 <= s.length <= 5 x 10^4s只包含字符 a,b 和 c 。
解题方法:滑动窗口
使用两个指针,时刻满足指针之间的"窗口"是最短的同时包含abc的窗口。
滑动方法一:枚举窗口起点l(l 到 r-1以及r的右边 全合法)------推荐方法二
两个指针,每次左指针 l l l右移一位,不断右移右指针 r r r直到窗口 [ l , r ) [l, r) [l,r)中含有三种字母。
那么对于起点 l l l,其到 r − 1 r-1 r−1合法到话,其到 r r r、 r + 1 r+1 r+1、 ⋯ \cdots ⋯、 n − 1 n-1 n−1(共计 n − r + 1 n-r+1 n−r+1个)都合法。
滑动方法二:枚举窗口终点r(l-1以及l-1的左边 到 r 全合法)
两个指针,每次右指针 r r r右移一位,不断右移左指针 l l l直到首次满足 窗口 l , r l, r l,r中不含有三种字母。
那么 l − 1 , r l-1, r l−1,r、 l − 2 , r l-2, r l−2,r、 ⋯ \cdots ⋯、 0 , r 0, r 0,r(如果存在)(共计l个)均合法。
时空复杂度分析
- 时间复杂度 O ( l e n ( s ) ) O(len(s)) O(len(s)),每个字母最多被遍历两次
- 空间复杂度 O ( 1 ) O(1) O(1)
AC代码
C++------方法二
cpp
/*
* @LastEditTime: 2026-07-01 15:38:27
*/
class Solution {
public:
int numberOfSubstrings(string s) {
int ans = 0;
int cnt[3] = {0};
for (int l = 0, r = 0, n = s.size(); r < n; r++) {
cnt[s[r] - 'a']++;
while (cnt[0] && cnt[1] && cnt[2]) {
cnt[s[l++] - 'a']--;
}
ans += l;
}
return ans;
}
};
C++------方法一
cpp
/*
* @LastEditTime: 2026-07-01 15:34:43
*/
class Solution {
public:
int numberOfSubstrings(string s) {
int ans = 0;
int cnt[3] = {0};
for (int l = 0, r = 0, n = s.size(); l < n; l++) {
while (r < n && !(cnt[0] && cnt[1] && cnt[2])) {
cnt[s[r++] - 'a']++;
}
// printf("l = %d, r = %d, cnt = {%d, %d, %d}, ans += %d\n", l, r, cnt[0], cnt[1], cnt[2], n - r + 1);
ans += (cnt[0] && cnt[1] && cnt[2]) ? n - r + 1 : 0;
cnt[s[l] - 'a']--;
}
return ans;
}
};
Python------方法二
python
'''
LastEditTime: 2026-07-01 15:49:11
'''
class Solution:
def numberOfSubstrings(self, s: str) -> int:
ans = 0
cnt = [0, 0, 0]
l = r = 0
for r in range(len(s)):
cnt[ord(s[r]) - ord('a')] += 1
while cnt[0] and cnt[1] and cnt[2]:
cnt[ord(s[l]) - ord('a')] -= 1
l += 1
ans += l
return ans
同步发文于CSDN和我的个人博客,原创不易,转载经作者同意后请附上原文链接哦~
千篇源码题解已开源