結果

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

ソースコード

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