如果你不是用DP而是用数学推导等方法做出的本题，一定要注意——本题的测试点是有 $p\le3$ 的情况的——被卡了好几次WA（悲）