[LeetCode] 395. 至少有 K 个重复字符的最长子串

[LeetCode] 395. 至少有 K 个重复字符的最长子串

题目

给你一个字符串 s 和一个整数 k ，请你找出 s 中的最长子串， 要求该子串中的每一字符出现次数都不少于 k 。返回这一子串的长度。

示例 1：

输入：s = "aaabb", k = 3
输出：3
解释：最长子串为 "aaa" ，其中 'a' 重复了 3 次。

思路

分治，在一个子串中，如果某一个字符的数量小于 k ，则这个子串的任意包含这个字符的子串都不满足要求，这时就可以按照这个字符将子串切分成若干段，然后再对这些段进行相同的处理。

代码

class Solution {
public:
    int dfs(const string& s, int l, int r, int k) {
        vector<int> cnt(26, 0);
        for (int i = l; i <= r; i++) {
            cnt[s[i] - 'a']++;
        }
        char split = 0;
        for (int i = 0; i < 26; i++) {
            if (cnt[i] > 0 && cnt[i] < k) {
                split = i + 'a';
                break;
            }
        }
        if (split == 0) {
            return r - l + 1;
        }
        int i = l;
        int ret = 0;
        while (i <= r) {
            while (i <= r && s[i] == split) {
                i++;
            }
            if (i > r) {
                break;
            }
            int start = i;
            while (i <= r && s[i] != split) {
                i++;
            }
            int length = dfs(s, start, i - 1, k);
            ret = max(ret, length);
        }
        return ret;
    }
    int longestSubstring(string s, int k) {
        return dfs(s, 0, s.size()-1, k);
    }
};