【滑动窗口】至多包含 K 个不同字符的最长子串

题目

340. 至多包含 K 个不同字符的最长子串

---

给你一个字符串 s 和一个整数 k ，请你找出 至多 包含 k 个 不同 字符的最长子串，并返回该子串的长度。

示例 1：

输入：s = "eceba", k = 2
输出：3
解释：满足题目要求的子串是 "ece" ，长度为 3 。

示例 2：

输入：s = "aa", k = 1
输出：2
解释：满足题目要求的子串是 "aa" ，长度为 2 。

提示：

• 1 <= s.length <= 5 * 10^4
• 0 <= k <= 50

解题

方法一：滑动窗口

思路

至多包含一个不同字符：【哈希表, 滑动窗口】无重复字符的最长子串
至多包含两个不同字符：【滑动窗口】至多包含两个不同字符的最长子串
那么显然，这题是以上两题的升级版，把判断收缩窗口时把固定数字换成 k 即可。

代码

class Solution {
public:
    int lengthOfLongestSubstringKDistinct(string s, int k) {
        unordered_map<char, int> window;
        int n = s.length();
        int left = 0, right = 0;
        int diff_cnt = 0, max_len = 0;
        while (right < n) {
            if (++window[s[right++]] == 1) ++diff_cnt;
            while (diff_cnt > k) {
                if (--window[s[left++]] == 0) --diff_cnt;
            }
            max_len = max(max_len, right - left);
        }
        return max_len;
    }
};