题目链接

题意

给你一个字符串 s 和一个整数 repeatLimit ，用 s 中的字符构造一个新字符串 repeatLimitedString ，使任何字母 连续 出现的次数都不超过 repeatLimit 次。你不必使用 s 中的全部字符。

返回 字典序最大的 repeatLimitedString 。

如果在字符串 a 和 b 不同的第一个位置，字符串 a 中的字母在字母表中出现时间比字符串 b 对应的字母晚，则认为字符串 a 比字符串 b 字典序更大 。如果字符串中前 min(a.length, b.length) 个字符都相同，那么较长的字符串字典序更大。
提示：

$1 <= repeatLimit <= s.length <= 10^5$
s 由小写英文字母组成

思路

贪心的构造

• 每次将当前剩余的字典序最大的字符添加到答案里，如果该字符已经连续出现了repeatLimit次，则先将当前剩余的字典序次大的字符添加到答案里，再继续将当前剩余的字典序最大的字符添加到答案里，直到该最大的字符用完或是没有次大的字符可以插入

有两种写法

1. 多层for循环，不断尝试填入新字符
   • 结合上述的思路可以得知，每个字符i最多被添加min(repeatLimit,mp[i])次，其中mp[i]为字符i的出现次数
   • 可以先统计字符串s里每个字符的出现次数
   • 倒序进行遍历，这时候i里维护的就是当前剩余的字典序最大的字符
   • 尝试将字符i添加到答案字符串里
   • 如果i已经填完了的话，跳出循环，找下一个当前剩余的字典序最大的字符
   • 否则的话找第一个存在且字典序次大的字符，插入到答案字符串里，这样后面就可以继续填字母i
2. 使用优先队列维护字典序最大字符和次大字符
   • 思路1的本质是通过for循环找当前剩余的字典序最大/次大的字符，可以通过优先队列来维护

代码

func repeatLimitedString(s string, repeatLimit int) string {
   
    mp := make(map[rune]int, 26)
    for _, ch := range s {
   
        mp[ch]++
    }
    ans := make([]rune,0,len(s))
    for i := 'z'; i >= 'a'; i-- {
   //倒序填入字母
        las := i - 1//记录次小的字母值
        for {
   
            for j := 0; j < repeatLimit && mp[i] > 0; j++ {
   //最多填入min(repeatLimit,mp[i])个字母i
                mp[i]--
                ans = append(ans, i)
            }
            if mp[i] == 0 {
   //i填完了 找下一个字典序最大的字母
                break
            }
            for las >= 0 && mp[las] == 0 {
   //找到第一个存在的字典序次大的字母
                las--
            }
            if las < 0 {
   //找不到 跳出
                break
            }
            //先填入次大字母 后面可以继续填最大字母i
            mp[las]--
            ans = append(ans, las)
        }
    }
    return string(ans)
}

class Solution {
   
    public:
        string repeatLimitedString(string s, int repeatLimit) {
   
            int mp[26];
            for(int i=0; i<26; i++) {
   
                mp[i]=0;
            }
            for(char ch:s) {
   
                mp[ch-'a']++;
            }
            string ans;
            for(int i=25; i>=0; i--) {
   
                int las = i-1;
                while(1) {
   
                    for(int j=0; j<repeatLimit&&mp[i]>0; j++) {
   
                        mp[i]--;
                        ans.push_back(i+'a');
                        //ans = ans + char(i+'a');
                    }
                    if(mp[i]==0) {
   
                        break;
                    }
                    while(las>=0&&mp[las]==0) {
   
                        las--;
                    }
                    if(las < 0) {
   
                        break;
                    }
                    mp[las]--;
                    ans.push_back(las+'a');
                //    ans = ans + char(las+'a');
                }
            }
            return ans;
        }
};