結果
問題 | No.828 全方位神童数 |
ユーザー | kopricky |
提出日時 | 2019-04-11 22:03:05 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 1,371 ms / 2,000 ms |
コード長 | 1,801 bytes |
コンパイル時間 | 382 ms |
コンパイル使用メモリ | 10,932 KB |
実行使用メモリ | 73,332 KB |
最終ジャッジ日時 | 2023-09-27 01:23:49 |
合計ジャッジ時間 | 32,034 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 17 ms
8,124 KB |
testcase_01 | AC | 18 ms
7,980 KB |
testcase_02 | AC | 17 ms
8,020 KB |
testcase_03 | AC | 69 ms
11,532 KB |
testcase_04 | AC | 30 ms
9,316 KB |
testcase_05 | AC | 63 ms
11,312 KB |
testcase_06 | AC | 27 ms
8,940 KB |
testcase_07 | AC | 24 ms
8,952 KB |
testcase_08 | AC | 27 ms
8,896 KB |
testcase_09 | AC | 21 ms
8,680 KB |
testcase_10 | AC | 49 ms
10,700 KB |
testcase_11 | AC | 20 ms
8,548 KB |
testcase_12 | AC | 52 ms
10,636 KB |
testcase_13 | AC | 476 ms
33,600 KB |
testcase_14 | AC | 1,048 ms
60,444 KB |
testcase_15 | AC | 342 ms
27,080 KB |
testcase_16 | AC | 495 ms
34,696 KB |
testcase_17 | AC | 396 ms
29,284 KB |
testcase_18 | AC | 731 ms
45,544 KB |
testcase_19 | AC | 1,337 ms
72,168 KB |
testcase_20 | AC | 858 ms
52,204 KB |
testcase_21 | AC | 1,137 ms
64,112 KB |
testcase_22 | AC | 245 ms
21,896 KB |
testcase_23 | AC | 78 ms
12,332 KB |
testcase_24 | AC | 668 ms
41,656 KB |
testcase_25 | AC | 257 ms
22,100 KB |
testcase_26 | AC | 1,245 ms
67,736 KB |
testcase_27 | AC | 1,048 ms
59,344 KB |
testcase_28 | AC | 1,272 ms
68,076 KB |
testcase_29 | AC | 1,179 ms
64,988 KB |
testcase_30 | AC | 661 ms
41,408 KB |
testcase_31 | AC | 803 ms
40,632 KB |
testcase_32 | AC | 1,205 ms
67,644 KB |
testcase_33 | AC | 1,336 ms
73,172 KB |
testcase_34 | AC | 1,070 ms
59,972 KB |
testcase_35 | AC | 694 ms
43,560 KB |
testcase_36 | AC | 862 ms
51,388 KB |
testcase_37 | AC | 560 ms
36,916 KB |
testcase_38 | AC | 785 ms
47,692 KB |
testcase_39 | AC | 1,257 ms
68,076 KB |
testcase_40 | AC | 552 ms
36,864 KB |
testcase_41 | AC | 454 ms
32,232 KB |
testcase_42 | AC | 1,371 ms
73,332 KB |
testcase_43 | AC | 827 ms
68,048 KB |
testcase_44 | AC | 314 ms
31,500 KB |
testcase_45 | AC | 472 ms
43,568 KB |
ソースコード
import sys import time 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): 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 return ver1 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): pp = i for v in G[i]: par = uf.find(v) graph[i].append(ver[par]) pp = uf.unite(par, pp) ver[uf.find(i)] = i dfs(n-1) ret = 1 for i in range(n): ret = ret * ans[i] % MOD print(ret)