結果

問題 No.1207 グラフX
ユーザー ああいい
提出日時 2022-01-27 17:40:47
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,336 ms / 2,000 ms
コード長 919 bytes
コンパイル時間 308 ms
コンパイル使用メモリ 82,548 KB
実行使用メモリ 332,420 KB
最終ジャッジ日時 2025-03-25 08:10:24
合計ジャッジ時間 36,553 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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()
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