代码中有两个函数,barnYard1() 用于输入数据并调用 solve2() 函数来找到可能的解决方案。solve2() 函数使用嵌套的循环来尝试不同的动物数量组合,并检查它们是否满足给定的头和腿的总数。
修改后:
def barnYard():
heads = int(input('Enter number of heads: '))
legs = int(input('Enter number of legs: '))
solutionFound = False
for spiders in range(heads + 1):
for chickens in range(heads - spiders + 1):
pigs = heads - chickens - spiders
total_legs = 4 * pigs + 2 * chickens + 8 * spiders
if total_legs == legs:
print('Number of pigs:', pigs)
print('Number of chickens:', chickens)
print('Number of spiders:', spiders)
solutionFound = True
break # 如果找到一个解决方案,退出循环
if solutionFound:
break # 如果找到一个解决方案,退出外部循环
if not solutionFound:
print('There is no solution')
barnYard()
|