結果

問題 No.3579 区間積逆像
コンテスト
ユーザー LyricalMaestro
提出日時 2026-07-05 16:08:04
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
WA  
実行時間 -
コード長 1,081 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 771 ms
コンパイル使用メモリ 96,052 KB
実行使用メモリ 113,076 KB
最終ジャッジ日時 2026-07-05 16:08:10
合計ジャッジ時間 5,039 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 28 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

# https://yukicoder.me/problems/no/3579


def main():
    N, P, C = map(int, input().split())
    A = list(map(int, input().split()))

    if C == 0:
        array = []
        prev = False
        for a in A:
            if a % P != 0:
                if not prev:
                    array.append(1)
                else:
                    array[-1] += 1
                prev = True
            else:
                prev = False
        ans = 0
        for x in array:
            ans += (x * (x + 1)) // 2
        
        total = (N * (N + 1)) // 2
        print(total - ans)
    else:
        answer = 0
        p_map = {1: 1}
        cum_prod_a = 1
        inv_c = pow(C, P - 2, P)
        for a in A:
            cum_prod_a *= a
            cum_prod_a %= P

            x = cum_prod_a * inv_c
            x %= P
            if x in p_map:
                answer += p_map[x]
            
            if cum_prod_a not in p_map:
                p_map[cum_prod_a] = 0
            p_map[cum_prod_a] += 1
        
        print(answer)



if __name__ == "__main__":
    main()
0