767. 重构字符串

767. 重构字符串

原始题目链接：https://leetcode-cn.com/problems/reorganize-string/

给定一个字符串S，检查是否能重新排布其中的字母，使得两相邻的字符不同。

若可行，输出任意可行的结果。若不可行，返回空字符串。

解题思路：

使用堆进行优化时间复杂度，python里面的堆是小根堆，堆顶元素是最小值，如果需要弹出最大值，可以将字典中统计的值取负号来反向堆，堆顶元素弹出后会自动构建小根堆，出堆操作后需要更新字符的值，使用一个元组进行基数操作（+1，因为是负数），在操作元组前，这里需要注意的是，先将上一个元组记录的信息入堆，即将会得到第二个数目最多的字符，这样确保是每次是不同的字符。在操作堆之前，可以先判断字符串S中出现最多的字符个数是否大于字符串长度的一半，超过半数肯定无法重构，具体实现看代码及注释。

代码实现：

class Solution:
    def reorganizeString(self, S: str) -> str:
        # 初始化重构的字符串
        res = ""
        
        # 统计S中各个字符的个数，chars_nums是一个字典
        chars_nums = collections.Counter(S)
        max_nums_char = chars_nums.most_common(1)
        
        # 如果最大的字符个数大于字符串的一半的长度，则一定不满足条件
        if max_nums_char[0][1] > (len(S) + 1) // 2:
            return res
            
        # 记录使用过的字符和个数，使用一次个数减1，key是个数，value是字符
        used_char = (0, None)
        
        # 使用堆这个数据结构优化时间复杂度，py的堆只有小根堆
        # 堆顶元素是最小元素，可以将字符出现的个数取负数
        # 每次堆顶元素出堆，保证取出的是出现最多字符来构建新字符串
        # 再从剩余的字符串中取出出现次数最多的字符，取出后再将上一次
        # 取出的字符个数减掉1，放入堆中，堆会根据入堆后自动构建小根堆
        # 再执行相同的步骤开始构建字符串，直到堆中元素全部取出

        # 构建小根堆，构建堆必须使用heappush内置方法，堆的元素是
        # 字符个数的负数和字符
        pq = []
        
        for k, v in chars_nums.items():
            heapq.heappush(pq, (-v, k))
            
        # 开始重构新的满足题目条件的字符串
        while pq:
            # 取出堆顶元素
            num, char = heapq.heappop(pq)
            res += char
            
            # 将更新使用过的字符（个数减1）入堆
            # 当个数为0时，即该字符全部使用完，不满足入堆条件
            # 所以小于0的才可以入堆进行更新操作
            # 注意和下一行代码的顺序，这样才能确保每次是不同的字符
            if used_char[0] < 0:
                heapq.heappush(pq, used_char)
                
            # 更新使用过的字符，个数减1
            used_char = (num + 1, char)
        
        return res