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

Category	Difficulty	Likes	Dislikes
algorithms	Medium (52.57%)	868	-

Tags

 

Companies

 

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

如果不存在这样的子字符串，则返回 0。

 

示例 1：

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

示例 2：

输入：s = "ababbc", k = 2
输出：5
解释：最长子串为 "ababb" ，其中 'a' 重复了 2 次， 'b' 重复了 3 次。

 

提示：

• 1 <= s.length <= 104
• s 仅由小写英文字母组成
• 1 <= k <= 105


 

解析：

10的4次方当然不能用dp，考虑O（n）的算法和O（nlogn）的，又看到s仅有小写字母构成，才26个，所以考虑与字母相关

三分法，找到最左边和最右边第一个不符合出现频率大于等于k的下标，然后只看这俩元素左边、右边和中间就好了

 

/*
 * @lc app=leetcode.cn id=395 lang=cpp
 *
 * [395] 至少有 K 个重复字符的最长子串
 */

// @lc code=start
class Solution {
public:

    int dfs(string &s, int &k, int x, int y) {
        if (x > y) return 0;
        int vis_word[26];
        memset(vis_word, 0, sizeof(vis_word));
        for (int i = x; i <= y; i++) {
            vis_word[s[i] - 'a']++;
        }
        for (int i = 0; i < 26; i++) {
            if (vis_word[i] >= k) vis_word[i] = 0;
        }
        int l = x, r = y;
        while(l <= y) {
            if (vis_word[s[l] - 'a']) {
                break;
            }
            l++;
        }
        while(r >= x) {
            if (vis_word[s[r] - 'a']) {
                break;
            }
            r--;
        }
        if (l > r) return y - x + 1;
        int len1 = dfs(s, k, x, l - 1);
        int len2 = dfs(s, k, r + 1, y);
        int len3 = dfs(s, k, l + 1, r - 1);
        return max(max(len1, len2), len3);
    }

    int longestSubstring(string s, int k) {
        return dfs(s, k, 0, s.length() - 1);
    }
};
// @lc code=end