結果
問題 | No.2344 (l+r)^2 |
ユーザー | titia |
提出日時 | 2023-06-11 02:01:48 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 618 ms / 2,000 ms |
コード長 | 1,463 bytes |
コンパイル時間 | 822 ms |
コンパイル使用メモリ | 87,072 KB |
実行使用メモリ | 112,808 KB |
最終ジャッジ日時 | 2023-08-31 01:40:44 |
合計ジャッジ時間 | 6,073 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 79 ms
75,888 KB |
testcase_01 | AC | 618 ms
78,476 KB |
testcase_02 | AC | 288 ms
79,040 KB |
testcase_03 | AC | 301 ms
80,064 KB |
testcase_04 | AC | 317 ms
79,056 KB |
testcase_05 | AC | 320 ms
81,092 KB |
testcase_06 | AC | 287 ms
78,944 KB |
testcase_07 | AC | 264 ms
88,360 KB |
testcase_08 | AC | 281 ms
89,916 KB |
testcase_09 | AC | 226 ms
85,868 KB |
testcase_10 | AC | 234 ms
86,584 KB |
testcase_11 | AC | 172 ms
82,892 KB |
testcase_12 | AC | 497 ms
112,808 KB |
ソースコード
# Lucasの定理 # 素数pと非負整数m, nに対して、Combi(m,n) mod pを求める。 # https://manabitimes.jp/math/1324 # modを取らない二項係数の計算は予め容易しておく。 # ここでの実装はパスカルの三角形を使ったもの。 Combi=[[] for i in range(100+1)] Combi[0]=[1,0] for i in range(1,100+1): Combi[i].append(1) for j in range(i): Combi[i].append(Combi[i-1][j]+Combi[i-1][j+1]) Combi[i].append(0) def p_rep(p,x): ANS=[] for i in range(100): ANS.append(x%p) x//=p if x==0: break return ANS def Combi_mod_p(m,n,p): ANS=1 A=p_rep(p,m) B=p_rep(p,n) for i in range(min(len(A),len(B))): ANS=ANS*Combi[A[i]][B[i]]%p return ANS def calc(A): B=[] for i in range(len(A)-1): B.append(((A[i]+A[i+1])**2)%(2**M)) return B T=int(input()) for tests in range(T): N,M=map(int,input().split()) A=list(map(int,input().split())) if N<=50: for tests in range(N-1): A=calc(A) print(A[0]) else: tt=50 X=[0]*tt Y=[Combi_mod_p(N-tt,i,2) for i in range(N-tt+1)] for i in range(tt): for j in range(N-tt+1): X[i]+=A[i+j]%2*Y[j] X[i]%=2 #print(A,Y,X,len(X)) A=X for tests in range(tt-1): A=calc(A) print(A[0])