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

发布时间 2024-01-05 11:46:56作者: WTSRUVF
CategoryDifficultyLikesDislikes
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