結果
問題 | No.828 全方位神童数 |
ユーザー | kopricky |
提出日時 | 2019-04-11 20:41:37 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 535 ms / 2,000 ms |
コード長 | 1,811 bytes |
コンパイル時間 | 785 ms |
コンパイル使用メモリ | 87,160 KB |
実行使用メモリ | 127,244 KB |
最終ジャッジ日時 | 2023-08-30 05:41:12 |
合計ジャッジ時間 | 16,978 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 70 ms
71,292 KB |
testcase_01 | AC | 72 ms
71,128 KB |
testcase_02 | AC | 74 ms
70,980 KB |
testcase_03 | AC | 184 ms
81,252 KB |
testcase_04 | AC | 129 ms
78,408 KB |
testcase_05 | AC | 168 ms
79,768 KB |
testcase_06 | AC | 115 ms
78,176 KB |
testcase_07 | AC | 108 ms
78,192 KB |
testcase_08 | AC | 119 ms
78,124 KB |
testcase_09 | AC | 85 ms
75,680 KB |
testcase_10 | AC | 166 ms
79,648 KB |
testcase_11 | AC | 84 ms
75,736 KB |
testcase_12 | AC | 166 ms
79,928 KB |
testcase_13 | AC | 283 ms
95,620 KB |
testcase_14 | AC | 421 ms
120,648 KB |
testcase_15 | AC | 264 ms
90,960 KB |
testcase_16 | AC | 305 ms
96,476 KB |
testcase_17 | AC | 285 ms
92,504 KB |
testcase_18 | AC | 375 ms
109,212 KB |
testcase_19 | AC | 498 ms
122,800 KB |
testcase_20 | AC | 395 ms
114,924 KB |
testcase_21 | AC | 449 ms
124,016 KB |
testcase_22 | AC | 235 ms
87,956 KB |
testcase_23 | AC | 195 ms
81,768 KB |
testcase_24 | AC | 360 ms
103,600 KB |
testcase_25 | AC | 249 ms
87,856 KB |
testcase_26 | AC | 524 ms
127,244 KB |
testcase_27 | AC | 468 ms
120,396 KB |
testcase_28 | AC | 499 ms
127,184 KB |
testcase_29 | AC | 494 ms
126,128 KB |
testcase_30 | AC | 353 ms
103,576 KB |
testcase_31 | AC | 343 ms
104,120 KB |
testcase_32 | AC | 499 ms
126,892 KB |
testcase_33 | AC | 535 ms
126,704 KB |
testcase_34 | AC | 461 ms
121,032 KB |
testcase_35 | AC | 367 ms
104,816 KB |
testcase_36 | AC | 403 ms
114,104 KB |
testcase_37 | AC | 327 ms
97,476 KB |
testcase_38 | AC | 384 ms
109,188 KB |
testcase_39 | AC | 503 ms
127,100 KB |
testcase_40 | AC | 339 ms
97,388 KB |
testcase_41 | AC | 304 ms
94,244 KB |
testcase_42 | AC | 532 ms
126,576 KB |
testcase_43 | AC | 220 ms
122,732 KB |
testcase_44 | AC | 140 ms
90,656 KB |
testcase_45 | AC | 171 ms
102,588 KB |
ソースコード
import sys MOD = 1000000007 class UnionFind: def __init__(self, node_size): self._node = node_size self.par = [i for i in range(self._node)] self.rank = [0] * self._node def find(self, ver): if self.par[ver] == ver: return ver else: self.par[ver] = self.find(self.par[ver]) return self.par[ver] def unite(self, ver1, ver2): ver1, ver2 = self.find(ver1), self.find(ver2) if ver1 == ver2: return if self.rank[ver1] < self.rank[ver2]: ver1, ver2 = ver2, ver1 self.par[ver2] = ver1 if self.rank[ver1] == self.rank[ver2]: self.rank[ver1] += 1 def same(self, ver1, ver2): return self.find(ver1) == self.find(ver2) def dfs(u): head = 1 stack[0], stack[1] = -1, u while head > 0: cur, prv, depth = stack[head], stack[head-1], head if visit[cur] == 0: ans[cur] += depth if visit[cur] == len(graph[cur]): head -= 1 continue head += 1 stack[head] = graph[cur][visit[cur]] visit[cur] += 1 if __name__ == '__main__': n = int(sys.stdin.readline()) G, graph = [[] for _ in range(n)], [[] for _ in range(n)] ver, visit, stack = [0]*n, [0]*n, [0]*(n+1) ans = list(map(int, sys.stdin.readline().split())) for i in range(n-1): u, v = map(int, sys.stdin.readline().split()) if u > v: u, v = v, u G[v-1].append(u-1) ver[i+1] = i+1 uf = UnionFind(n) for i in range(n): for v in G[i]: graph[i].append(ver[uf.find(v)]), uf.unite(i, v) ver[uf.find(i)] = i dfs(n-1) ret = 1 for i in range(n): ret = ret * ans[i] % MOD print(ret)