题目来源

• leetcode：358. K 距离间隔重排字符串

题目描述

class Solution {
    
public:
    string rearrangeString(string str, int k) {
    
      
    }
};

题目解析

思路：

• 准备：
  • 先统计所有字符出现的次数，存入一个map中（key = char， value = count）
  • 然后使用map按照重复次数从大到小构建一个大根堆
  • 创建一个string ans对象用来表示结果
  • 创建一个queue来同步已经加入到ans中的字符
• 循环遍历大根堆的字符
  • 先插入的是数字最多的字符，插入之后将这个字符的数量-1，然后把这个字符移到queue中（因为下一次不能插入它了，必须等到长度到达k之后才能插入它）
  • 然后判断queue中的元素是否为k，如果是的话，说明队头的元素可以再次尝试去插入了，因此将它放入到大根堆里面
• 最后，当大根堆为空时：
  • 如果ans.size == ori.size说明重构完成，返回ans
  • 如果ans.size != ori.size，说明还有些字符挂在queue中，重构失败，返回""

class Solution {
    
public:
    string rearrangeString(string str, int k) {
    
        if(k == 0){
    
            return str;
        }
        int N = str.size();
        std::string ans;
        std::unordered_map<char, int> m;
        // 遍历字符，统计字符的出现次数
        for(auto c : str){
    
            m[c]++;
        }
        // 装入大顶堆，按照字符重复次数作为比较
        std::priority_queue<pair<char, int>, std::vector<pair<char, int>>,
                cmp> maxHeap(m.begin(), m.end());
        std::queue<pair<char, int>> queue;
        while (!maxHeap.empty()){
    
            auto curr = maxHeap.top(); maxHeap.pop();// 从大顶堆取出重复次数最多的字符
            ans += curr.first;
            curr.second--;// 用掉一个字符，次数减一
            queue.push(curr);// 放入到queue中，因为k距离后还要用
            if(queue.size() == k){
    
                // queue的大小到达了k，也就是说我们已经越过了k个单位，在结果中应该要出现相同的字母了
                auto f = queue.front(); queue.pop();
                if(f.second > 0){
    
                    // 该字符的重复次数大于 0，则添加入大顶堆中，要是0那还加它干嘛
                    maxHeap.push(f);
                }
            }
        }

        return ans.size() == str.size() ? ans : "";
    }
};

类似题目

• leetcode：358. K 距离间隔重排字符串
• leetcode：621. Task Scheduler
• leetcode：767. 重构字符串（堆）