結果
問題 | No.1207 グラフX |
ユーザー | anagohirame |
提出日時 | 2020-08-30 18:23:51 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,233 ms / 2,000 ms |
コード長 | 1,306 bytes |
コンパイル時間 | 159 ms |
コンパイル使用メモリ | 82,032 KB |
実行使用メモリ | 322,692 KB |
最終ジャッジ日時 | 2024-04-27 13:00:16 |
合計ジャッジ時間 | 28,384 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 555 ms
147,720 KB |
testcase_01 | AC | 554 ms
147,432 KB |
testcase_02 | AC | 544 ms
148,244 KB |
testcase_03 | AC | 611 ms
148,372 KB |
testcase_04 | AC | 578 ms
147,756 KB |
testcase_05 | AC | 1,212 ms
320,460 KB |
testcase_06 | AC | 1,181 ms
322,692 KB |
testcase_07 | AC | 1,170 ms
322,536 KB |
testcase_08 | AC | 565 ms
129,572 KB |
testcase_09 | AC | 572 ms
133,736 KB |
testcase_10 | AC | 1,233 ms
278,264 KB |
testcase_11 | AC | 1,188 ms
322,648 KB |
testcase_12 | AC | 552 ms
128,624 KB |
testcase_13 | AC | 350 ms
104,740 KB |
testcase_14 | AC | 600 ms
147,300 KB |
testcase_15 | AC | 574 ms
134,692 KB |
testcase_16 | AC | 362 ms
102,812 KB |
testcase_17 | AC | 521 ms
120,512 KB |
testcase_18 | AC | 376 ms
115,276 KB |
testcase_19 | AC | 480 ms
118,064 KB |
testcase_20 | AC | 583 ms
147,492 KB |
testcase_21 | AC | 107 ms
81,544 KB |
testcase_22 | AC | 476 ms
120,892 KB |
testcase_23 | AC | 465 ms
125,760 KB |
testcase_24 | AC | 348 ms
111,372 KB |
testcase_25 | AC | 603 ms
148,236 KB |
testcase_26 | AC | 515 ms
129,492 KB |
testcase_27 | AC | 577 ms
140,440 KB |
testcase_28 | AC | 570 ms
137,460 KB |
testcase_29 | AC | 553 ms
140,416 KB |
testcase_30 | AC | 357 ms
104,672 KB |
testcase_31 | AC | 317 ms
101,800 KB |
testcase_32 | AC | 297 ms
103,628 KB |
testcase_33 | AC | 333 ms
103,856 KB |
testcase_34 | AC | 558 ms
132,488 KB |
testcase_35 | AC | 133 ms
82,072 KB |
testcase_36 | AC | 554 ms
136,880 KB |
testcase_37 | AC | 511 ms
124,700 KB |
testcase_38 | AC | 226 ms
89,000 KB |
testcase_39 | AC | 327 ms
105,036 KB |
testcase_40 | AC | 206 ms
95,668 KB |
testcase_41 | AC | 459 ms
113,244 KB |
testcase_42 | AC | 35 ms
54,152 KB |
testcase_43 | AC | 33 ms
52,464 KB |
testcase_44 | AC | 37 ms
53,008 KB |
testcase_45 | AC | 512 ms
147,924 KB |
testcase_46 | AC | 511 ms
147,776 KB |
testcase_47 | AC | 516 ms
148,136 KB |
testcase_48 | AC | 519 ms
148,076 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)