改正了runtime error的问题，没考虑p==1的情况,提交ac代码