結果

問題 No.329 全射
コンテスト
ユーザー Tawara
提出日時 2015-12-25 17:17:58
言語 PyPy2
(7.3.15)
結果
TLE  
実行時間 -
コード長 913 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,801 ms
コンパイル使用メモリ 76,288 KB
実行使用メモリ 145,036 KB
最終ジャッジ日時 2024-09-18 23:52:21
合計ジャッジ時間 6,022 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 TLE * 1 -- * 29
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from heapq import heappop, heappush
from collections import defaultdict as ddict
p = 10**9+7
N,M = map(int,raw_input().split())
w = map(int,raw_input().split())
w_max = max(w)
E = [[] for i in xrange(N)]
for t in xrange(M):
	i,j = map(lambda x:int(x)-1,raw_input().split())
	E[i].append(j)
C = [[0] for i in xrange(w_max+1)]
C[0].extend([1,0])
for i in xrange(1,w_max+1):
	C[i].extend([(C[i-1][j]+C[i-1][j+1])%p for j in xrange(i+1)]+[0])
F = [ddict(int) for i in xrange(w_max+1)]
for s in xrange(N):
	ws = w[s]
	V = ddict(int)
	Q = []
	heappush(Q,(-w[s],s))
	while Q:
		w_min,h = heappop(Q)
		if w_min >= V[h]: continue
		V[h] = w_min
		if w[h] <= -w_min: F[ws][w[h]] += 1
		w_min = max(w_min,-w[h])
		for nxt in E[h]: heappush(Q,(w_min,nxt))
ans = 0
for wi in xrange(1,w_max+1):
	for wj,v in F[wi].iteritems():
		ans += v*sum(C[wj][k+1]*pow(wj-k,wi,p)*((-1)**(k%2)) for k in xrange(wj)) % p
		ans %= p
print ans
0