結果

問題 No.2709 1975 Powers
コンテスト
ユーザー graythunder1
提出日時 2024-08-24 01:53:18
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 661 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 167 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 18,208 KB
最終ジャッジ日時 2024-08-24 01:53:23
合計ジャッジ時間 4,575 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 1 TLE * 1 -- * 23
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

N, P, Q = map(int, input().split(" "))
A = sorted(list(map(int, input().split(" "))))

def mod_pow(a, n):
    res = 1
    while n > 0:
        if n & 1:
            res = res * a % P
        a = a * a % P
        n >>= 1
    return res


dval_set = [set([mod_pow(5, A[-1])])]
for i in range(1, N):
    dval_set.append(dval_set[i-1] | set([mod_pow(5, A[N-i-1])]))
dval_set = dval_set[::-1]

ans = 0
for ai in range(N-3):
    for bi in range(ai+1, N-2):
        for ci in range(bi+1, N-1):
            abcval = (mod_pow(10, A[ai]) + mod_pow(9, A[bi]) + mod_pow(7, A[ci])) % P
            if (Q - abcval) % P in dval_set[ci+1]:
                ans += 1
print(ans)
0