結果

問題 No.1354 Sambo's Treasure
ユーザー titia
提出日時 2021-09-01 23:04:25
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 2,054 bytes
コンパイル時間 186 ms
コンパイル使用メモリ 82,496 KB
実行使用メモリ 397,764 KB
最終ジャッジ日時 2024-11-28 18:51:27
合計ジャッジ時間 71,365 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25 WA * 15 TLE * 21
権限があれば一括ダウンロードができます

ソースコード

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(key=itemgetter(0))
T.sort(key=itemgetter(0))

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:
        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]

    #print(LIST)

    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]
                    NDP[l+1]%=mod

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

        #print(NDP)
                
        if j!=len(LIST)-1:
            for l in range(1,j):
                NDP[l]-=NDP[l+1]
        else:            
            for l in range(j):
                NDP[l]-=NDP[l+1]

        #print(NDP)

        DP[j]=NDP

    #print(DP)

    dp=DP[-1]

    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]
            NEXT[i+j]%=mod

    NOW=NEXT

#print(NOW)
print(sum(NOW[:K+1])%mod)
0