結果
問題 | No.1025 Modular Equation |
ユーザー |
![]() |
提出日時 | 2020-04-10 23:43:10 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 589 bytes |
コンパイル時間 | 99 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 21,704 KB |
最終ジャッジ日時 | 2024-09-16 03:11:14 |
合計ジャッジ時間 | 8,230 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | WA * 6 TLE * 1 -- * 25 |
ソースコード
# coding: utf-8# Your code here!import syssys.setrecursionlimit(10**6)readline = sys.stdin.readlineread = sys.stdin.readp,n,k,b,*a = [int(i) for i in read().split()]lst = [0]*pfor i in range(p): lst[pow(i,k,p)] += 1MOD = 10**9+7x = 1M = 1<<96NUM = 96*pALL = (1<<NUM)-1m = mm = ((1<<32)-1)<<64for _ in range(n):m <<= 96m ^= mmpa = (1 << 64) - ((1 << 64) % MOD)for ai in a:y = 0for i,c in enumerate(lst):y += c<<(i*ai%p*96)x = x*yx = (x>>NUM) + (x&ALL)x -= ((x & m) >> 64) * paans = (x>>(b*96))&((1<<96)-1)print(ans%MOD)