358. K 距離間隔重排字符串

題目：

給你一個非空的字符串 s 和一個整數 k，你要將這個字符串中的字母進行重新排列，使得重排后的字符串中相同字母的位置間隔距離至少為 k。

所有輸入的字符串都由小寫字母組成，如果找不到距離至少為 k 的重排結果，請返回一個空字符串 ""。

示例 1：

輸入: s = "aabbcc", k = 3
輸出: "abcabc"
解釋: 相同的字母在新的字符串中間隔至少 3 個單位距離。
示例 2:

輸入: s = "aaabc", k = 3
輸出: ""
解釋: 沒有辦法找到可能的重排結果。
示例 3:

輸入: s = "aaadbbcc", k = 2
輸出: "abacabcd"
解釋: 相同的字母在新的字符串中間隔至少 2 個單位距離。

 

解答：

貪心法，每次盡量取頻次大的字母加入結果。

先統計每個字母出現的次數，存入map

然后把所有的[字母：頻次]存入最大堆

每次從最大堆中取k個最大頻次的字母，加入結果字符串。然后相應頻次減一后再放回最大堆。

最終要么堆為空，則成功。如果堆中剩下不到k個字母，那么只能取一次（即字符串結尾，因為后面沒有其他元素了），否則失敗。

class Solution {
public:
    string rearrangeString(string s, int k) {
        if(k==0){
            return s;
        }
        vector<int> cnt(26,0);
        for(char& c:s){
            cnt[c-'a']+=1;
        }
        priority_queue<pair<int,char>> max_heap;
        for(int i=0;i<26;++i){
            if(cnt[i]>0){
                max_heap.push(make_pair(cnt[i],'a'+i));
            }
        }
        string res;
        while(res.size()<s.size()){
            vector<pair<int,char>> tmp;
            int min_len=min(k,int(max_heap.size()));
            for(int i=0;i<min_len;++i){
                auto cur=max_heap.top();
                max_heap.pop();
                res+=cur.second;
                if(--cur.first>0){
                    tmp.push_back(cur);
                }
            }
            if(min_len<k and res.size()!=s.size()){
                return "";
            }
            for(auto& pai:tmp){
                max_heap.push(pai);
            }
        }
        return res;
    }
};

時間O(n log n)，空間O(n)