leetcode_1358 包含所有三种字符的子字符串数目

626 阅读1分钟

要求

给你一个字符串 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^4
  • s 只包含字符 a,b 和 c 。

核心代码

class Solution:
    def numberOfSubstrings(self, s: str) -> int:
        if len(s) <= 2:
            return 0
        length = len(s)
        left,right = 0,2
        ans = 0

        while left < length - 2:
            window = s[left:right + 1]
            if "a" in window and "b" in window and "c" in window:
                ans += length - right
                left += 1
            else:
                right += 1
                if right == length:
                    break
        return ans

image.png

解题思路:双指针 滑动窗口

  1. 如果窗口内有 a, b, c, 那么窗口继续向右拉开都满足
  2. 如果窗口内没有 a, b, c, 那么 right 指针右移找到满足窗口内有 a, b, c
  3. 然后 left 指针右移,每次都要判断上面两种情况

自己的理解:我们取[left:right + 1]的长度的窗口,在这个窗口中我们,已经存在了abc,那么从right一直到length的位置上都是满足条件的,所以ans直接加上length - right, 然后左指针右移一次;要不然右指针右移,当我们的右指针移动到length的时候,还不满足,及不可能又满足的了,所以直接跳出循环。