結果

問題 No.1354 Sambo's Treasure
ユーザー titia
提出日時 2021-09-01 23:32:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 392 ms / 2,000 ms
コード長 2,176 bytes
コンパイル時間 162 ms
コンパイル使用メモリ 82,424 KB
実行使用メモリ 141,440 KB
最終ジャッジ日時 2024-11-28 20:34:11
合計ジャッジ時間 14,116 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 61
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
from operator import itemgetter

N,M,L,K=map(int,input().split())
C=[tuple(map(int,input().split())) for i in range(M)]
T=[tuple(map(int,input().split())) for i in range(L)]
mod=998244353

C.append((N,N))

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

FACT_INV=[pow(FACT[-1],mod-2,mod)]
for i in range(5*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

C.sort()
T.sort()

for i in range(1,M+1):
    if C[i][1]<C[i-1][1]:
        print(0)
        exit()
        
T2=[[] for i in range(M+1)]

ind=0

C.append((0,0))

for i in range(M+1):
    x,y=C[i]

    while ind<L and (T[ind][0]<x or (T[ind][0]==x and T[ind][1]<=y)):
        if C[i-1][1]<=T[ind][1]<=C[i][1]:
            T2[i].append(T[ind])
        ind+=1

NOW=[1]

for i in range(M+1):
    sx,sy=C[i-1]
    gx,gy=C[i]

    LIST=[(sx,sy)]+T2[i]+[(gx,gy)]
    DP=[[] for i in range(len(LIST))]
    DP[0]=[1]

    for j in range(1,len(LIST)):
        tx,ty=LIST[j]
        
        NDP=[0]*(j+1)

        for k in range(j-1,-1,-1):
            x,y=LIST[k]

            COM=Combi((tx-x)+(ty-y),(tx-x))

            if j!=len(LIST)-1:

                for l in range(k+1):
                    NDP[l+1]+=COM*DP[k][l]%mod
                    NDP[l+1]%=mod

            else:
                for l in range(k+1):
                    NDP[l]+=COM*DP[k][l]%mod
                    NDP[l]%=mod
                
        if j!=len(LIST)-1:
            for l in range(1,j):
                NDP[l]-=NDP[l+1]
                NDP[l]%=mod
        else:            
            for l in range(j):
                NDP[l]-=NDP[l+1]
                NDP[l]%=mod

        DP[j]=NDP

    dp=DP[-1]

    while dp[-1]==0:
        dp.pop()

    NEXT=[0]*(len(dp)+len(NOW))

    for i in range(len(dp)):
        for j in range(len(NOW)):
            NEXT[i+j]+=dp[i]*NOW[j]%mod
            NEXT[i+j]%=mod

    NOW=NEXT

    while NOW[-1]==0:
        NOW.pop()

ANS=0
for i in range(min(len(NOW),K+1)):
    ANS+=NOW[i]
    ANS%=mod
    
print(ANS)
0