Time-1 - Coconuts of Palm Tree

この問題は 3重ループ文 を問います。

問題文は、相異なる $3$ つの整数 $i,j,k\ (1\le i\lt j\lt k\le N)$ を用いて次のように表せます。

• $A_i+A_j+A_k$ が $M$ の倍数であるような $(i,j,k)$ の選び方はいくつ存在するか？

よって、これは3重ループを用いて実装することができます。したがって計算量は $O(N^3)$ で実装が可能です。
またそれぞれ $A_i \bmod M$ ※を計算することで $O(N^2\log_{}N)$ で実装することもできます。以下が解答例です(C++,Python)。

※ $A\bmod B$ は $A$ を $B$ で割った余りを表します

• C++

• Python