結果

問題 No.1207 グラフX
ユーザー 37zigen
提出日時 2020-08-30 22:01:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,650 ms / 2,000 ms
コード長 1,089 bytes
コンパイル時間 287 ms
コンパイル使用メモリ 82,592 KB
実行使用メモリ 292,524 KB
最終ジャッジ日時 2025-03-14 05:25:22
合計ジャッジ時間 40,025 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(10**6)

MOD=10**9+7

class UnionFind():
    def __init__(self,n):
        self.upper=[-1]*n

    def root(self,x):
        if self.upper[x]<0:
            return x
        else:
            self.upper[x]=self.root(self.upper[x])
            return self.upper[x]

    def equiv(self,x,y):
        return self.root(x)==self.root(y)

    def unite(self,x,y):
        x=self.root(x)
        y=self.root(y)
        if x==y:
            return
        if self.upper[x]>self.upper[y]:
            x,y=y,x
        self.upper[x]+=self.upper[y]
        self.upper[y]=x

N,M,X=map(int,input().split())
g=[[] for _ in range(N+1)]
uf=UnionFind(N+1)
for i in range(M):
    x,y,z=map(int,input().split())
    if not uf.equiv(x,y):
        g[x].append((y,pow(X,z,MOD)))
        g[y].append((x,pow(X,z,MOD)))
        uf.unite(x,y)

sz=[1]*(N+1)
ans=0

def dfs(v,p):
    global ans
    for dst,cost in g[v]:
        if dst==p:
            continue
        dfs(dst,v)
        ans+=sz[dst]*(N-sz[dst])%MOD*cost%MOD
        ans%=MOD
        sz[v]+=sz[dst]

dfs(1,-1)
print(ans)
0