解説P=N×PP=N\times PP=N×P を変形すると (N−1)P=0(N-1)P=0(N−1)P=0 となる。したがって、等式が成り立つのは N=1N=1N=1 または P=0P=0P=0 の場合に限られる。 主解法ではこの二つの条件を調べる。愚直解法では N×PN\times PN×P を直接計算し、PPP と比較する。どちらも各ケースの時間・追加空間計算量は O(1)O(1)O(1)、全体の時間計算量は O(T)O(T)O(T) であり、すべての制約で利用できる。 ∣NP∣≤106|NP|\leq 10^6∣NP∣≤106、∣(N−1)P∣≤1 001 000|(N-1)P|\leq 1\,001\,000∣(N−1)P∣≤1 なので、符号付き 32 ビット整数でも安全に計算できる。 の場合があるため、無条件に両辺を で割ってはいけない。 Solution written by GPT