第二回日本最強プログラマー学生選手権 D Nowhere P
$ A_1
が決まってしまえば,
$ A_2, A_3, A_4, ...
は
$ A_1, A_2, A_3, ...
以外の数であることがわかり, またこれらは均等なので, 答えは
$ (P-1)(P-2)^{N-1}
通りとなる.
実装例:
https://atcoder.jp/contests/jsc2021/submissions/21810840