結果

問題 No.3579 区間積逆像
コンテスト
ユーザー titia
提出日時 2026-07-09 03:06:13
言語 PyPy3
(8.0.0 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 91 ms / 2,000 ms
+ 479µs
コード長 994 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 82 ms
コンパイル使用メモリ 64,908 KB
実行使用メモリ 89,520 KB
最終ジャッジ日時 2026-10-23 13:51:39
合計ジャッジ時間 4,345 ms
ジャッジサーバーID
（参考情報） 		judge4_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
input = sys.stdin.readline

from collections import Counter

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

Z=[]
for i in range(N):
    if A[i]==0:
        Z.append(i)

ANS=0
if C==0:
    ind=0

    for i in range(N):
        if ind==len(Z):
            break
        if Z[ind]<i:
            ind+=1
        if ind==len(Z):
            break

        ANS+=N-Z[ind]

    print(ANS)

else:
    LIST=[]
    X=[]
    for i in range(N):
        if A[i]==0:
            if X!=[]:
                LIST.append(X)
                X=[]
            else:
                X=[]
        else:
            X.append(A[i])

    if X!=[]:
        LIST.append(X)

    #print(LIST)

    for L in LIST:
        S=[1]
        now=1
        for x in L:
            now=now*x%P
            S.append(now)

        CA=Counter(S)

        for s in S:
            CA[s]-=1

            need=C*s%P

            ANS+=CA[need]

            #print(s,ANS,need)

    print(ANS)
0