358. Rearrange String k Distance Apart

這道題就是把所有值都進行最大區(qū)分，在重新加入heap的時候要注意，如果兩個值有同樣的count，那么后訪問過的要先加進去，比如說bba, size=1，先加入b，此時a,b的count都是1，這時候要先加入a，再加入b

class Solution(object):
    def rearrangeString(self, s, k):
        """
        :type s: str
        :type k: int
        :rtype: str
        """
        if k == 0:
            return s
        h = {}
        for c in s:
            h[c] = h.get(c, 0) + 1
        
        heap = []
        for key, val in h.iteritems():
            heapq.heappush(heap, [-val, key])
        
        res = []
        
        while heap:
            temp = []
            n = len(s) - len(res)
            for _ in range(min(k, n)):
                if not heap:
                    return ""
                cur_val, key = heapq.heappop(heap)
                res.append(key)
                cur_val += 1
                if cur_val != 0:
                    temp.append([cur_val, key])
            
            while temp:
                heapq.heappush(heap, temp.pop())
        
        return "".join(res)