結果
| 問題 |
No.2499 Sum of Products of Sums
|
| コンテスト | |
| ユーザー |
とりゐ
|
| 提出日時 | 2023-09-05 12:03:00 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
MLE
|
| 実行時間 | - |
| コード長 | 1,339 bytes |
| コンパイル時間 | 970 ms |
| コンパイル使用メモリ | 82,304 KB |
| 実行使用メモリ | 851,460 KB |
| 最終ジャッジ日時 | 2024-06-23 07:40:23 |
| 合計ジャッジ時間 | 3,072 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 MLE * 1 |
| other | -- * 17 |
ソースコード
# TLE or RE
mod=998244353
table_size=2*10**5
fac=[1]*(table_size+1)
finv=[1]*(table_size+1)
for i in range(2,table_size+1):
fac[i]=fac[i-1]*i%mod
finv[table_size]=pow(fac[table_size],mod-2,mod)
for i in range(table_size-1,-1,-1):
finv[i]=finv[i+1]*(i+1)%mod
def rebuild(n):
global table_size,fac,finv
fac+=[0]*(n-table_size)
fac+=[0]*(n-table_size)
finv+=[0]*(n-table_size)
for i in range(table_size+1,n+1):
fac[i]=fac[i-1]*i%mod
finv[n]=inv(fac[n])
for i in range(n-1,table_size,-1):
finv[i]=finv[i+1]*(i+1)%mod
table_size=n
def binom(n,k):
if n<0 or k<0:
return 0
if k>n:
return 0
if n>table_size:
rebuild(n+10**4)
return (fac[n]*finv[k]%mod)*finv[n-k]%mod
def fpow(x,k):
res=1
while k:
if k&1:
res=res*x%mod
x=x*x%mod
k>>=1
return res
def inv(a):
if a<table_size:
return fac[a-1]*finv[a]%mod
return fpow(a,mod-2)
def f(N,S,k):
# |X| = N, sum X <= S なる X に対する Π[i<=k] X_i の総和
return binom(N+S,N+k)
H,W=map(int,input().split())
dp=[0]*(W+1)
dp[0]=1
for _ in range(H):
L,R=map(int,input().split())
res=[0]*(W+1)
for i in range(W+1):
res[i]=f(W,R,i)-f(W,L-1,i)
ndp=[0]*(W+1)
for i in range(W+1):
for j in range(i,W+1):
ndp[j]+=dp[i]*res[j-i]%mod*binom(W-i,j-i)
ndp[j]%=mod
dp=ndp
print(dp[W])
とりゐ