結果

問題 No.329 全射
コンテスト
ユーザー titia
提出日時 2026-06-27 01:25:01
言語 PyPy3
(8.0.1 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,281 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 99 ms
コンパイル使用メモリ 64,380 KB
実行使用メモリ 86,496 KB
最終ジャッジ日時 2026-10-22 12:35:10
合計ジャッジ時間 23,280 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 40 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
input = sys.stdin.readline

from heapq import heappop,heappush

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

mod=10**9+7

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

E=[[] for i in range(N)]

for i in range(M):
    x,y=list(map(int,input().split()))
    x-=1
    y-=1
    E[x].append(y)

ANS=0

for i in range(N):
    DP=[0]*N
    DP[i]=W[i]

    Q=[]
    Q.append((-W[i],i))

    while Q:
        weight,ind=heappop(Q)

        weight=-weight

        for to in E[ind]:
            necw=min(weight,W[to])

            if DP[to]<necw:
                DP[to]=necw
                heappush(Q,(-DP[to],to))

    #print(DP)

    for j in range(N):
        if DP[j]==W[j]:
            score=0
            now=W[j]

            for k in range(W[j],0,-1):
                if k%2==W[j]%2:
                    score+=pow(k,W[i],mod)*Combi(W[j],k)
                else:
                    score-=pow(k,W[i],mod)*Combi(W[j],k)

            ANS+=score

print(ANS%mod)
0