用Python爱因斯坦数学题

在Python中解决爱因斯坦（Einstein）著名的数学谜题通常涉及到一些有趣的逻辑和条件判断题目。比如"不动脑筋的汉斯"（Hans and the Stairs）问题，这是一种典型的“囚徒困境”（Prisoner's Dilemma）类型的题目，也可以通过编程来模拟。

例如，你可以编写一个程序，让两个人（Hans和朋友）可以选择上楼梯的不同步数，而每一步都有奖励或惩罚。如果他们都选择相同的步数，他们都会得到一定的积分；但如果一个人选择更多，他可能会得到更多的积分，但另一个人会受到惩罚。这种问题是关于合作和自私之间的权衡，类似于爱因斯坦对人类理性的思考。

以下是简单的Python代码示例：

def stairs_game(steps, partner_steps):
    if steps == partner_steps:
        return "Both got the same reward"
    elif steps > partner_steps:
        return f"Hans gets a bigger reward, but his friend pays the price ({steps} vs {partner_steps})"
    else:
        return f"His friend gets a bigger reward ({partner_steps} vs {steps})"

# 示例
hans_steps = int(input("How many steps does Hans take? "))
friend_steps = int(input("How many steps does his friend take? "))
print(stairs_game(hans_steps, friend_steps))