結果
問題 | No.940 ワープ ε=ε=ε=ε=ε=│;p>д<│ |
ユーザー | maspy |
提出日時 | 2019-12-03 10:07:10 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 684 ms / 5,000 ms |
コード長 | 1,708 bytes |
コンパイル時間 | 198 ms |
コンパイル使用メモリ | 10,780 KB |
実行使用メモリ | 208,196 KB |
最終ジャッジ日時 | 2023-08-18 23:40:19 |
合計ジャッジ時間 | 8,128 ms |
ジャッジサーバーID (参考情報) |
judge10 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 117 ms
29,824 KB |
testcase_01 | AC | 118 ms
29,840 KB |
testcase_02 | AC | 117 ms
29,784 KB |
testcase_03 | AC | 122 ms
30,708 KB |
testcase_04 | AC | 118 ms
29,808 KB |
testcase_05 | AC | 117 ms
30,160 KB |
testcase_06 | AC | 115 ms
29,988 KB |
testcase_07 | AC | 118 ms
29,940 KB |
testcase_08 | AC | 116 ms
29,936 KB |
testcase_09 | AC | 114 ms
30,068 KB |
testcase_10 | AC | 116 ms
30,036 KB |
testcase_11 | AC | 117 ms
29,916 KB |
testcase_12 | AC | 114 ms
30,152 KB |
testcase_13 | AC | 116 ms
30,072 KB |
testcase_14 | AC | 117 ms
29,844 KB |
testcase_15 | AC | 165 ms
41,504 KB |
testcase_16 | AC | 261 ms
71,004 KB |
testcase_17 | AC | 398 ms
113,264 KB |
testcase_18 | AC | 446 ms
128,572 KB |
testcase_19 | AC | 387 ms
112,984 KB |
testcase_20 | AC | 524 ms
147,900 KB |
testcase_21 | AC | 292 ms
79,404 KB |
testcase_22 | AC | 508 ms
143,944 KB |
testcase_23 | AC | 262 ms
72,492 KB |
testcase_24 | AC | 505 ms
146,284 KB |
testcase_25 | AC | 634 ms
195,968 KB |
testcase_26 | AC | 684 ms
208,196 KB |
ソースコード
import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines import numpy as np X,Y,Z = map(int,read().split()) MOD = 10 ** 9 + 7 def cumprod(arr,MOD): L = len(arr); Lsq = int(L**.5+1) arr = np.resize(arr,Lsq**2).reshape(Lsq,Lsq) for n in range(1,Lsq): arr[:,n] *= arr[:,n-1]; arr[:,n] %= MOD for n in range(1,Lsq): arr[n] *= arr[n-1,-1]; arr[n] %= MOD return arr.ravel()[:L] def make_fact(U,MOD): x = np.arange(U,dtype=np.int64); x[0] = 1 fact = cumprod(x,MOD) x = np.arange(U,0,-1,dtype=np.int64); x[0] = pow(int(fact[-1]),MOD-2,MOD) fact_inv = cumprod(x,MOD)[::-1] return fact,fact_inv def make_power(a,L,MOD): B = L.bit_length() x = np.empty(1 + (1<<B),np.int64) x[0] = 1; x[1] = a for n in range(B): x[1<<n:1<<(n+1)] = x[:1<<n] * (a * x[(1<<n)-1] % MOD) % MOD return x[:L] U = 2 * (X + Y + Z + 100) fact,fact_inv = make_fact(U,MOD) power2 = make_power(2,U,MOD) power2_inv = make_power((1+MOD)//2,U,MOD) # P = 1/2(1-x)(1-y)(1-z) # (2P-1)^nを足しまくる # 1-(2P-1)^N / (2-2P), NはX+Y+Zより十分大きければなんでも N = X + Y + Z + 1 if not N&1: N += 1 A = fact[N] * fact_inv[:N+1] % MOD * fact_inv[:N+1][::-1] % MOD A *= power2[:N+1]; A %= MOD A[1::2] *= (-1) A[0] += 1 # 2-2x で割る A *= (MOD+1)//2; A %= MOD np.cumsum(A,out=A); A %= MOD # 1/(1-x)(1-y)(1-z)になおして A *= power2_inv[:N+1]; A %= MOD for x in [X,Y,Z]: # (1/1-T)^n でX次 B = np.zeros(N+1,np.int64) if x == 0: B[0] = 1 B[1:] = fact[x:x+N] * fact_inv[x] % MOD * fact_inv[:N] % MOD A *= B; A %= MOD answer = A.sum() % MOD print(answer)