Python鸡兔同笼代码

以下是一个Python的鸡兔同笼问题的代码示例：

def calculate_chicken_rabbit(total_heads, total_legs):
    # 鸡兔同笼问题的解法
    for num_chicken in range(total_heads + 1):
        num_rabbit = total_heads - num_chicken
        if (2 * num_chicken + 4 * num_rabbit) == total_legs:
            return num_chicken, num_rabbit
    return None

# 输入总头数和总腿数
total_heads = int(input("请输入总头数："))
total_legs = int(input("请输入总腿数："))

result = calculate_chicken_rabbit(total_heads, total_legs)
if result:
    num_chicken, num_rabbit = result
    print("鸡的数量为：", num_chicken)
    print("兔的数量为：", num_rabbit)
else:
    print("无法确定鸡兔的数量。")

这段代码通过遍历可能的鸡的数量，计算出对应的兔的数量，并检查是否满足总腿数的条件。如果找到了满足条件的鸡和兔的数量，就输出结果；否则，输出无法确定鸡兔数量的提示。