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

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

 

def balloon_arrange(data):

# 获取气球总数
total = 0
for i in data.values():
    total = total + i
# 气球颜色最多的数量
sort_balloon = sorted(data.items(), key=lambda x: x[1], reverse=True)
max_count = sort_balloon[0][1]
if max_count > (total + 1) // 2:
    return "无法做到相同颜色的气球不挂在一起"

res = []
list = [None] * total
for k, count in sort_balloon:
    res.extend(k * count)
if max_count == (total + 1) // 2:
    list[::2], list[1::2] = res[:(total + 1) // 2], res[(total + 1) // 2:]
else:
    list[::2], list[1::2] = res[total // 2:], res[:total // 2]
return "-".join(list)
data = { 'r': 2, 'b': 4, 'w': 1, }

print(balloon_arrange(data))

分析:

颜色最多的气球n个，其他气球总数大于等于 n-1

来源：力扣（LeetCode）