python编写羊车门问题