leetcode堆

最新推荐文章于 2024-03-21 20:14:07 发布

紫砂痕

最新推荐文章于 2024-03-21 20:14:07 发布

阅读量76

点赞数

分类专栏： python 基础 leetcode 文章标签： python leetcode

本文链接：https://blog.csdn.net/qq_36047533/article/details/115435828

版权

leetcode 同时被 2 个专栏收录

22 篇文章 0 订阅

订阅专栏

python 基础

11 篇文章 0 订阅

订阅专栏

1. leetcode 1054 距离相等的条形码

在这里插入图片描述


```python
import collections
class Solution:
    def rearrangeBarcodes(self, barcodes: List[int]) -> List[int]:
        if len(barcodes) == 1:
            return barcodes
        res = []
        dicta = collections.defaultdict(lambda:0)
        for s in barcodes:
            dicta[s] +=1   
        dicta[0] = 0    # 确保每一组数据都有差值 
        sort_dicta = sorted(dicta.items(),  key = lambda x:x[1])   
        while len(sort_dicta)> 1:
            i = len(sort_dicta) -1
            temp = sort_dicta[i][1]
            temp_dicta = dict(sort_dicta)
            j = i-1 
            res_temp = [sort_dicta[i][0]]
            while j >= 0:
                if sort_dicta[j][1] == temp:
                    res_temp.append(sort_dicta[j][0])
                    j-=1
                else:
                    if len(res_temp) == 1:
                        if sort_dicta[j][0]>0:
                            res_temp.append(sort_dicta[j][0])
                        diff = 1
                    else:
                        diff = temp - sort_dicta[j][1]
                    break
            res += res_temp*diff
            if j ==0:
                break
            else:
                for key in res_temp:   
                    temp_dicta[key] -=diff
                    if temp_dicta[key] ==0:
                        del temp_dicta[key] 
            sort_dicta = sorted(temp_dicta.items(),  key = lambda x:x[1])         
        return res