제출 #1364722

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1364722		zzzzzzzzzzzzzzz	Dubai Chewy Cookie (KAISTRUN26SPRING_C)	Pypy 3	100 / 100	226 ms	266376 KiB

Main

import sys
input=sys.stdin.readline
mod=998244353
n,m,q=map(int,input().split())
p=list(map(int,input().split()))
ch=[0]*(n+1)
ch[1]=1;ch[2]=2
l1=[]
l2=[]
l3=[]
for i in range(m):
    u,v=map(int,input().split())
    if(u<=2):ch[v]+=u
    if(v<=2):ch[u]+=v
for i in range(1,n+1):
    if(ch[i]==1):l1.append(p[i-1])
    if(ch[i]==2):l2.append(p[i-1])
    if(ch[i]==3):l3.append(p[i-1])
p1=[0]*2001
p2=[0]*2001
p3=[0]*2001
p1[0]=p2[0]=p3[0]=1
for i in range(len(l1)):
    x=l1[i];p0=p1[:]
    p1[0]=p0[0]*(1-x)%mod
    for j in range(1,2001):p1[j]=(p0[j-1]*x+p0[j]*(1-x))%mod
for i in range(len(l2)):
    x=l2[i];p0=p2[:]
    p2[0]=p0[0]*(1-x)%mod
    for j in range(1,2001):p2[j]=(p0[j-1]*x+p0[j]*(1-x))%mod
for i in range(len(l3)):
    x=l3[i];p0=p3[:]
    p3[0]=p0[0]*(1-x)%mod
    for j in range(1,2001):p3[j]=(p0[j-1]*x+p0[j]*(1-x))%mod
for i in range(q):
    s,t=map(int,input().split())
    ans=0
    for j in range(min(s,t)+1):
        ans+=p3[j]*p1[s-j]*p2[t-j]
        ans%=mod
    print(ans)

컴파일 시 표준 출력 (stdout) 메시지

Compiling 'Main.py'...

=======
  adding: __main__.pyc (deflated 46%)

=======

• 서브태스크 1: 20 / 20
• 서브태스크 2: 20 / 20
• 서브태스크 3: 20 / 20
• 서브태스크 4: 40 / 40