結果
問題 | No.1207 グラフX |
ユーザー |
|
提出日時 | 2020-08-30 14:44:47 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,653 ms / 2,000 ms |
コード長 | 1,556 bytes |
コンパイル時間 | 358 ms |
コンパイル使用メモリ | 82,772 KB |
実行使用メモリ | 346,700 KB |
最終ジャッジ日時 | 2025-03-31 04:49:46 |
合計ジャッジ時間 | 36,518 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 46 |
ソースコード
import sys input = lambda : sys.stdin.readline().rstrip() sys.setrecursionlimit(max(1000, 10**9)) write = lambda x: sys.stdout.write(x+"\n") # inputna n,m,X = list(map(int, input().split())) M = 10**9+7 xyz = [list(map(int, input().split())) for _ in range(m)] for i in range(m): xyz[i] = (xyz[i][0]-1, xyz[i][1]-1, xyz[i][2]) xyz.sort(key=lambda x: x[2]) class UnionFindTree: def __init__(self, n): self.n = n self.parent = list(range(n)) self.size = [1] * n def root(self, i): inter = set() while self.parent[i]!=i: inter.add(i) i = self.parent[i] r = i for i in inter: self.parent[i] = r return r def connect(self, i, j): ri = self.root(i) rj = self.root(j) if ri==rj: return if self.size[ri]<self.size[rj]: self.parent[ri] = rj self.size[rj] += self.size[ri] else: self.parent[rj] = ri self.size[ri] += self.size[rj] ans = 0 uf = UnionFindTree(n) ss = [0] * n es = [[] for _ in range(n)] for x,y,z in xyz: rx = uf.root(x) ry = uf.root(y) if rx==ry: continue uf.connect(x,y) es[x].append((y,z)) es[y].append((x,z)) ans = 0 size = [None]*n start = 0 def sub(u, p): global ans res = 0 for v,z in es[u]: if v==p: continue size = sub(v,u) res += size ans = ans + size * (n-size) * pow(X,z,M) ans %= M return res+1 sub(0,-1) print(ans%M)