結果

問題 No.2709 1975 Powers
ユーザー Cecil
提出日時 2024-03-31 14:31:55
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 942 bytes
コンパイル時間 394 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 19,620 KB
最終ジャッジ日時 2024-09-30 19:39:25
合計ジャッジ時間 3,724 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other TLE * 1 -- * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict as dd

def pow(x, n, mod):
    ans = 1
    #n が 0 になるまで計算を続ける
    while n:
        if n % 2:
            ans *= x
            ans %= mod
        x *= x
        n >>= 1
    return ans%mod

N,P,Q = map(int, input().split())
A = list(map(int, input().split()))
A = list(set(A))
A.sort()
N = len(A)
A_div = [ [0 for i in range(4)] for _ in range(N)] 
num = [10,9,7,5]
#cnt = dd(int)
for i in range(N):
    for j in range(4):
        A_div[i][j] = pow(num[j], A[i], P)
ans = 0
for i in range(N):
    for j in range(i, N):
        if i==j:
            continue
        for k in range(j, N):
            if j==k:
                continue
            for l in range(k, N):
                if k==l:
                    continue
                val = A_div[i][0]+A_div[j][1]+A_div[k][2]+A_div[l][3]
                val %= P
                if val == Q:
                    ans += 1
print(ans)
0