前段时间面试,有一家公司要给一个题的解法,自己也在网上找找了类似的题目,其核心思想可以参考 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 --> 蓝红蓝红白红