結果

問題 No.1207 グラフX
ユーザー ああいい
提出日時 2022-01-27 17:46:49
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 789 ms / 2,000 ms
コード長 1,172 bytes
コンパイル時間 380 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 171,488 KB
最終ジャッジ日時 2024-12-25 19:36:10
合計ジャッジ時間 31,523 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(10 ** 8)

N,M,X = map(int,input().split())
G = [[] for _ in range(N+1)]
P = 10 ** 9 + 7
edge = []
parent = list(range(N+1))
def find(i):
    if parent[i] == i:return i
    parent[i] = find(parent[i])
    return parent[i]
def unite(i,j):
    I = find(i)
    J = find(j)
    if I == J:return False
    parent[I] = J
    parent[i] = J
    return True
def isSame(i,j):
    return find(i) == find(j)

for _ in range(M):
    x,y,z = map(int,input().split())
    if isSame(x,y):
        continue
    else:
        unite(x,y)
        G[x].append(y)
        G[y].append(x)
        edge.append((x,y,z))
child = [1] * (N + 1)
"""
def dfs(now = 1,p = 0):
    for v in G[now]:
        if v == p:continue
        dfs(v,now)
        child[now] += child[v]
dfs()
"""
stack = [(1,0)]
order = []
while stack:
    now,p = stack.pop()
    order.append((now,p))
    for v in G[now]:
        if v == p:continue
        stack.append((v,now))
for v,p in order[::-1]:
    if p == 0:continue
    child[p] += child[v]
    
ans = 0
for x,y,z in edge:
    if child[x] > child[y]:
        x,y = y,x
    ans = (ans + child[x] * (N-child[x]) * pow(X,z,P)) % P
print(ans)
0