結果
問題 | No.1207 グラフX |
ユーザー | anagohirame |
提出日時 | 2020-08-30 18:23:51 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,589 ms / 2,000 ms |
コード長 | 1,306 bytes |
コンパイル時間 | 376 ms |
コンパイル使用メモリ | 82,004 KB |
実行使用メモリ | 322,744 KB |
最終ジャッジ日時 | 2024-12-21 07:52:00 |
合計ジャッジ時間 | 40,917 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 764 ms
147,904 KB |
testcase_01 | AC | 746 ms
147,612 KB |
testcase_02 | AC | 751 ms
148,144 KB |
testcase_03 | AC | 744 ms
148,868 KB |
testcase_04 | AC | 758 ms
148,036 KB |
testcase_05 | AC | 1,514 ms
320,512 KB |
testcase_06 | AC | 1,577 ms
322,744 KB |
testcase_07 | AC | 1,560 ms
322,392 KB |
testcase_08 | AC | 794 ms
129,632 KB |
testcase_09 | AC | 783 ms
134,024 KB |
testcase_10 | AC | 1,589 ms
277,112 KB |
testcase_11 | AC | 1,573 ms
322,576 KB |
testcase_12 | AC | 782 ms
129,040 KB |
testcase_13 | AC | 488 ms
105,168 KB |
testcase_14 | AC | 831 ms
147,840 KB |
testcase_15 | AC | 799 ms
134,376 KB |
testcase_16 | AC | 504 ms
103,064 KB |
testcase_17 | AC | 662 ms
121,120 KB |
testcase_18 | AC | 534 ms
115,140 KB |
testcase_19 | AC | 661 ms
118,132 KB |
testcase_20 | AC | 807 ms
147,692 KB |
testcase_21 | AC | 139 ms
81,780 KB |
testcase_22 | AC | 660 ms
121,228 KB |
testcase_23 | AC | 672 ms
125,764 KB |
testcase_24 | AC | 469 ms
111,060 KB |
testcase_25 | AC | 815 ms
147,968 KB |
testcase_26 | AC | 691 ms
129,176 KB |
testcase_27 | AC | 811 ms
140,768 KB |
testcase_28 | AC | 804 ms
137,128 KB |
testcase_29 | AC | 742 ms
140,932 KB |
testcase_30 | AC | 508 ms
104,864 KB |
testcase_31 | AC | 436 ms
101,956 KB |
testcase_32 | AC | 425 ms
103,680 KB |
testcase_33 | AC | 471 ms
104,064 KB |
testcase_34 | AC | 768 ms
132,428 KB |
testcase_35 | AC | 179 ms
81,844 KB |
testcase_36 | AC | 747 ms
136,904 KB |
testcase_37 | AC | 708 ms
124,544 KB |
testcase_38 | AC | 313 ms
89,344 KB |
testcase_39 | AC | 459 ms
105,092 KB |
testcase_40 | AC | 290 ms
95,876 KB |
testcase_41 | AC | 626 ms
113,408 KB |
testcase_42 | AC | 44 ms
52,096 KB |
testcase_43 | AC | 42 ms
52,608 KB |
testcase_44 | AC | 43 ms
52,352 KB |
testcase_45 | AC | 708 ms
148,096 KB |
testcase_46 | AC | 703 ms
148,480 KB |
testcase_47 | AC | 691 ms
148,012 KB |
testcase_48 | AC | 699 ms
148,172 KB |
ソースコード
import sys sys.setrecursionlimit(10**6) def input(): return sys.stdin.buffer.readline()[:-1] MOD = 10**9+7 class UnionFind(): def __init__(self, size): self.table = [-1 for _ in range(size)] self.size = [1 for _ in range(size)] def find(self, x): while self.table[x] >= 0: if self.table[self.table[x]] >= 0: self.table[x] = self.table[self.table[x]] x = self.table[x] return x def unite(self, x, y): s1 = self.find(x) s2 = self.find(y) if s1 != s2: if self.table[s1] > self.table[s2]: self.table[s2] = s1 self.table[s1] -= 1 self.size[s1] += self.size[s2] self.size[s2] = 0 else: self.table[s1] = s2 self.table[s2] -= 1 self.size[s2] += self.size[s1] self.size[s1] = 0 return n, m, X = map(int, input().split()) edge = [] for _ in range(m): x, y, z = map(int, input().split()) edge.append((x-1, y-1, z)) edge.sort(key=lambda x: x[2]) uf = UnionFind(n) adj = [[] for _ in range(n)] for x, y, z in edge: if uf.find(x) != uf.find(y): uf.unite(x, y) adj[x].append((y, z)) adj[y].append((x, z)) ans = 0 num = [1 for _ in range(n)] def dfs(i, p): global ans for j, z in adj[i]: if j == p: continue dfs(j, i) ans += num[j] * (n - num[j]) * pow(X, z, MOD) ans %= MOD num[i] += num[j] return dfs(0, -1) print(ans)