結果

問題 No.2164 Equal Balls
ユーザー titiatitia
提出日時 2022-12-15 09:25:53
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,217 bytes
コンパイル時間 197 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 173,856 KB
最終ジャッジ日時 2024-04-26 11:53:59
合計ジャッジ時間 8,971 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 84 ms
97,664 KB
testcase_01 AC 92 ms
99,072 KB
testcase_02 AC 111 ms
108,288 KB
testcase_03 AC 89 ms
99,584 KB
testcase_04 AC 95 ms
103,424 KB
testcase_05 AC 93 ms
105,728 KB
testcase_06 AC 100 ms
106,240 KB
testcase_07 AC 93 ms
103,424 KB
testcase_08 TLE -
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 -- -
testcase_47 -- -
testcase_48 -- -
testcase_49 -- -
testcase_50 -- -
testcase_51 -- -
testcase_52 -- -
testcase_53 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline

N,M=map(int,input().split())
A=list(map(int,input().split()))
B=list(map(int,input().split()))

mod=998244353

FACT=[1]
for i in range(1,2*10**5+1):
    FACT.append(FACT[-1]*i%mod)

FACT_INV=[pow(FACT[-1],mod-2,mod)]
for i in range(2*10**5,0,-1):
    FACT_INV.append(FACT_INV[-1]*i%mod)

FACT_INV.reverse()

def Combi(a,b):
    if 0<=b<=a:
        return FACT[a]*FACT_INV[b]%mod*FACT_INV[a-b]%mod
    else:
        return 0

PLUS=[[1]*601 for i in range(M)]

for i in range(N):
    k=i%M
    a=A[i]
    b=B[i]

    for j in range(601):
        p=j-300
        # bの方がaよりp多い

        score=0

        for x in range(a+1):
            y=x+p

            if 0<=y<=b:
                score+=Combi(a,x)*Combi(b,y)
                score%=mod
                
        PLUS[k][j]=PLUS[k][j]*score%mod

DP=[0]*(2*300*300+1)
DP[300*300]=1

for i in range(M):
    NDP=[0]*(2*300*300+1)

    for j in range(2*300*300+1):
        if DP[j]==0:
            continue
        for k in range(601):
            p=k-300

            if 0<=j+p<2*300*300:
                NDP[j+p]=(NDP[j+p]+DP[j]*PLUS[i][k])%mod
    DP=NDP

print(DP[300*300])

            
        
    

        
    
0