第61题:正方形拼接
中等
题目描述:
现在有一堆木棒,告诉你它们的长度,判断能否用这些木棒拼接成正方形。 注意:所有的木棒都要用上,且不能截断。 给你一个正整数list L, 如 L=[1,1,1,1], L中的每个数字代表一个木棒的长度,如果这些 木棒能够拼成一个正方形,输出Yes,否则输出No。 如L=[1,1,1,1],则输出Yes;L=[1,1,1],则输出No。 注:数据已于2014-03-11加强,之前通过的代码可能无法再次通过
示例:
输入:L = [1, 1, 1, 1]
输出:Yes
def is_square(L): #判断是否可以组成正方形
#判断总长度是不是四的倍数
sum_list = sum(L)
if sum_list % 4 != 0 or not sum_list: #正整数的list
return 'No'
#若可以组成四的倍数
edge = int(sum_list / 4) #边的长度
if max(L) > edge:
return 'No'
edge_list = [edge,edge,edge,edge]
L.sort(reverse = True) #翻转
for i in range(len(L)):
index = edge_list.index(max(edge_list) ) #找到最大值的下标
edge_list[index] = edge_list[index] - L[i] #加入木棒
if not min(edge_list):
return 'Yes'
else:
return 'No'
print(is_square(L))