气球排列

43次阅读

共计 720 个字符,预计需要花费 2 分钟才能阅读完成。

前段时间面试,有一家公司要给一个题的解法,自己也在网上找找了类似的题目,其核心思想可以参考 LeetCode 中的 767 重构字符串, 我就不在这里做过多的介绍了。链接如下:
重构字符串

题目描述:
小明有一些气球想挂在墙上装饰 他希望相同颜色的气球不要挂在一起 写一个算法帮他得出一种可行的挂气球方式 自行定义函数 输入和返回 如果无法做到相同颜色的气球不挂在一起 请定义合适的异常方式返回

class BalloonArrangement(object):
    """气球排列"""

    @staticmethod
    def cal_ret(data: dict):
        '''
        计算出一种可行的挂气球方式
        :param data:
        :return:
        '''

        bal_count = sum([i for i in data.values()])  # 气球总数
        # 气球颜色最多的数量
        max_count = sorted(data.items(), key=lambda v: v[1], reverse=True)[0][1]
        res = []
        if max_count > (bal_count + 1) // 2:
            # 无法做到相同颜色的气球不挂在一起
            return ''
        for k, count in data.items():
            res.extend(k * count)

        ans = [None] * bal_count
        ans[::2], ans[1::2] = res[bal_count//2:], res[:bal_count//2]
        return "-".join(ans)
data = {
    'r': 3,  # 红色气球 3 个
    'b': 2,  # 蓝色 2 个
    'w': 1,  # 白色 1 个
}
b = BalloonArrangement.cal_ret(data)
print(b)  # b-r-b-r-w-r  --> 蓝红蓝红白红

正文完
 0