結果
| 問題 | No.1321 塗るめた |
| ユーザー |
 37zigen
|
| 提出日時 | 2020-12-18 03:15:29 |
| 言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 1,054 bytes |
| コンパイル時間 | 670 ms |
| コンパイル使用メモリ | 12,112 KB |
| 実行使用メモリ | 122,944 KB |
| 最終ジャッジ日時 | 2023-10-21 07:48:06 |
| 合計ジャッジ時間 | 7,817 ms |
|
ジャッジサーバーID （参考情報） |
judge10 / judge11 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 1,359 ms
118,104 KB |
| testcase_01 | TLE | - |
| testcase_02 | -- | - |
| testcase_03 | -- | - |
| testcase_04 | -- | - |
| testcase_05 | -- | - |
| testcase_06 | -- | - |
| testcase_07 | -- | - |
| testcase_08 | -- | - |
| testcase_09 | -- | - |
| 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 | -- | - |
| testcase_35 | -- | - |
| testcase_36 | -- | - |
| testcase_37 | -- | - |
| testcase_38 | -- | - |
| testcase_39 | -- | - |
| testcase_40 | -- | - |
| testcase_41 | -- | - |
| testcase_42 | -- | - |
| testcase_43 | -- | - |
| testcase_44 | -- | - |
| testcase_45 | -- | - |
| testcase_46 | -- | - |
ソースコード
import numpy as np
p=998244353
def mul(a, b, MOD=p):
fft=np.fft
def submul(f,g):
L=len(f)+len(g)-1
return fft.irfft(fft.rfft(f,L)*fft.rfft(g,L),L)%MOD
al=a&(1<<15)-1
bl=b&(1<<15)-1
ah=a>>15
bh=b>>15
x,y,z=submul(al,bl),submul(ah,bh),submul(al+ah,bl+bh)
x,y,z=map(lambda v:(v+.5).astype(np.int64),[x,y,z])
return (x+(y<<30)+((z-x-y)<<15))%MOD
fac=[1, 1]
inv=[1, 1]
ifac=[1, 1]
for i in range(2, int(2e5)):
fac.append(fac[-1]*i%p)
inv.append(p-inv[p%i]*(p//i)%p)
ifac.append(ifac[-1]*inv[i]%p)
def fpow(a, n):
if n==0:
return np.array([1])
else:
ret=fpow(mul(a, a)[:10**5+10], n//2)[:10**5+10]
if n%2==1:
ret=mul(a, ret)[:10**5+10]
return ret
def comb(n, k):
return fac[n]*ifac[n-k]%p*ifac[k]%p
N,M,K=map(int,input().split())
f=np.array([0]*(10**5+10))
for i in range(len(f)):
f[i]=ifac[i]
f[0]=0
f=fpow(f,K)
ans=0
for i in range(K,N+1):
ans+=comb(N, i)*comb(M, K)%p*pow(M,N-i,p)%p*f[i]%p*fac[i]%p
ans%=p
print(ans)