結果

問題 No.1025 Modular Equation
ユーザー convexineqconvexineq
提出日時 2020-04-10 23:43:10
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 589 bytes
コンパイル時間 165 ms
コンパイル使用メモリ 10,984 KB
実行使用メモリ 18,396 KB
最終ジャッジ日時 2023-10-14 08:02:48
合計ジャッジ時間 8,375 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 17 ms
7,940 KB
testcase_01 AC 17 ms
7,816 KB
testcase_02 AC 29 ms
8,256 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 TLE -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

# coding: utf-8
# Your code here!
import sys
sys.setrecursionlimit(10**6)
readline = sys.stdin.readline
read = sys.stdin.read


p,n,k,b,*a = [int(i) for i in read().split()]

lst = [0]*p
for i in range(p): lst[pow(i,k,p)] += 1

MOD = 10**9+7

x = 1
M = 1<<96
NUM = 96*p
ALL = (1<<NUM)-1
m = mm = ((1<<32)-1)<<64
for _ in range(n):
    m <<= 96
    m ^= mm
pa = (1 << 64) - ((1 << 64) % MOD)

for ai in a:
    y = 0
    for i,c in enumerate(lst):
        y += c<<(i*ai%p*96)
    x = x*y
    x = (x>>NUM) + (x&ALL)
    x -= ((x & m) >> 64) * pa

ans = (x>>(b*96))&((1<<96)-1)
print(ans%MOD)
0