Allen Chou says to YSITD
沒有更 Pythonic 的解法嗎