結果
問題 | No.1207 グラフX |
ユーザー | rlangevin |
提出日時 | 2023-02-02 00:40:44 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,324 ms / 2,000 ms |
コード長 | 1,871 bytes |
コンパイル時間 | 282 ms |
コンパイル使用メモリ | 87,276 KB |
実行使用メモリ | 246,684 KB |
最終ジャッジ日時 | 2023-09-14 09:59:35 |
合計ジャッジ時間 | 44,312 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 889 ms
231,520 KB |
testcase_01 | AC | 886 ms
231,920 KB |
testcase_02 | AC | 895 ms
238,824 KB |
testcase_03 | AC | 889 ms
230,668 KB |
testcase_04 | AC | 895 ms
231,180 KB |
testcase_05 | AC | 1,321 ms
210,268 KB |
testcase_06 | AC | 1,310 ms
220,776 KB |
testcase_07 | AC | 1,314 ms
211,236 KB |
testcase_08 | AC | 812 ms
170,136 KB |
testcase_09 | AC | 838 ms
182,896 KB |
testcase_10 | AC | 1,324 ms
208,400 KB |
testcase_11 | AC | 1,312 ms
207,060 KB |
testcase_12 | AC | 782 ms
173,700 KB |
testcase_13 | AC | 514 ms
116,356 KB |
testcase_14 | AC | 904 ms
217,520 KB |
testcase_15 | AC | 831 ms
184,192 KB |
testcase_16 | AC | 520 ms
116,408 KB |
testcase_17 | AC | 665 ms
149,612 KB |
testcase_18 | AC | 564 ms
166,204 KB |
testcase_19 | AC | 678 ms
136,540 KB |
testcase_20 | AC | 884 ms
207,528 KB |
testcase_21 | AC | 160 ms
82,636 KB |
testcase_22 | AC | 688 ms
157,420 KB |
testcase_23 | AC | 703 ms
180,632 KB |
testcase_24 | AC | 525 ms
141,316 KB |
testcase_25 | AC | 924 ms
204,952 KB |
testcase_26 | AC | 781 ms
174,240 KB |
testcase_27 | AC | 874 ms
188,844 KB |
testcase_28 | AC | 850 ms
185,940 KB |
testcase_29 | AC | 839 ms
206,008 KB |
testcase_30 | AC | 533 ms
132,432 KB |
testcase_31 | AC | 465 ms
110,428 KB |
testcase_32 | AC | 450 ms
133,028 KB |
testcase_33 | AC | 506 ms
130,856 KB |
testcase_34 | AC | 828 ms
177,392 KB |
testcase_35 | AC | 205 ms
84,180 KB |
testcase_36 | AC | 819 ms
199,680 KB |
testcase_37 | AC | 743 ms
178,672 KB |
testcase_38 | AC | 340 ms
100,300 KB |
testcase_39 | AC | 499 ms
135,808 KB |
testcase_40 | AC | 345 ms
100,180 KB |
testcase_41 | AC | 674 ms
141,260 KB |
testcase_42 | AC | 73 ms
71,304 KB |
testcase_43 | AC | 71 ms
71,316 KB |
testcase_44 | AC | 72 ms
71,288 KB |
testcase_45 | AC | 857 ms
246,028 KB |
testcase_46 | AC | 853 ms
246,544 KB |
testcase_47 | AC | 854 ms
246,684 KB |
testcase_48 | AC | 859 ms
246,588 KB |
ソースコード
import sys readline = sys.stdin.readline class UnionFind(object): def __init__(self, n=1): self.par = [i for i in range(n)] self.rank = [0 for _ in range(n)] self.size = [1 for _ in range(n)] def find(self, x): if self.par[x] == x: return x else: self.par[x] = self.find(self.par[x]) return self.par[x] def union(self, x, y): x = self.find(x) y = self.find(y) if x != y: if self.rank[x] < self.rank[y]: x, y = y, x if self.rank[x] == self.rank[y]: self.rank[x] += 1 self.par[y] = x self.size[x] += self.size[y] def is_same(self, x, y): return self.find(x) == self.find(y) def get_size(self, x): x = self.find(x) return self.size[x] def non_rec_dfs(s): stack = [] stack.append(~s) stack.append(s) par = [-1] * N sz = [1] * N global ans while stack: u = stack.pop() if u >= 0: stack.append(~u) for v in G[u]: if v == par[u]: continue par[v] = u stack.append(v) else: u = ~u sz[par[u]] += sz[u] if par[u] != -1: ans += dic[(u, par[u])] * sz[u] * (N - sz[u]) ans %= mod N, M, X = map(int, input().split()) mod = 10 ** 9 + 7 U = UnionFind(N) D = [] for i in range(M): x, y, z = map(int, input().split()) x, y = x - 1, y - 1 D.append((z, x, y)) D.sort() G = [[] for i in range(N)] dic = dict() for z, x, y in D: if U.is_same(x, y): continue U.union(x, y) G[x].append(y) G[y].append(x) v = pow(X, z, mod) dic[(x, y)] = v dic[(y, x)] = v ans = 0 non_rec_dfs(0) print(ans)