結果
問題 | No.828 全方位神童数 |
ユーザー | kopricky |
提出日時 | 2019-04-11 22:06:32 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 1,331 ms / 2,000 ms |
コード長 | 1,793 bytes |
コンパイル時間 | 95 ms |
コンパイル使用メモリ | 10,788 KB |
実行使用メモリ | 73,476 KB |
最終ジャッジ日時 | 2023-08-30 05:43:20 |
合計ジャッジ時間 | 27,059 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 21 ms
8,088 KB |
testcase_01 | AC | 16 ms
8,104 KB |
testcase_02 | AC | 17 ms
8,200 KB |
testcase_03 | AC | 64 ms
11,692 KB |
testcase_04 | AC | 30 ms
9,232 KB |
testcase_05 | AC | 56 ms
11,288 KB |
testcase_06 | AC | 26 ms
8,944 KB |
testcase_07 | AC | 23 ms
8,952 KB |
testcase_08 | AC | 24 ms
8,820 KB |
testcase_09 | AC | 19 ms
8,748 KB |
testcase_10 | AC | 46 ms
10,724 KB |
testcase_11 | AC | 20 ms
8,756 KB |
testcase_12 | AC | 48 ms
10,744 KB |
testcase_13 | AC | 421 ms
33,688 KB |
testcase_14 | AC | 968 ms
60,340 KB |
testcase_15 | AC | 310 ms
26,972 KB |
testcase_16 | AC | 458 ms
34,520 KB |
testcase_17 | AC | 345 ms
29,228 KB |
testcase_18 | AC | 662 ms
45,384 KB |
testcase_19 | AC | 1,235 ms
72,280 KB |
testcase_20 | AC | 792 ms
52,132 KB |
testcase_21 | AC | 1,037 ms
64,356 KB |
testcase_22 | AC | 213 ms
21,852 KB |
testcase_23 | AC | 70 ms
12,372 KB |
testcase_24 | AC | 611 ms
41,404 KB |
testcase_25 | AC | 230 ms
22,260 KB |
testcase_26 | AC | 1,202 ms
67,972 KB |
testcase_27 | AC | 1,023 ms
59,644 KB |
testcase_28 | AC | 1,194 ms
68,152 KB |
testcase_29 | AC | 1,143 ms
65,036 KB |
testcase_30 | AC | 601 ms
41,504 KB |
testcase_31 | AC | 596 ms
40,576 KB |
testcase_32 | AC | 1,161 ms
67,808 KB |
testcase_33 | AC | 1,305 ms
73,080 KB |
testcase_34 | AC | 995 ms
59,880 KB |
testcase_35 | AC | 634 ms
43,564 KB |
testcase_36 | AC | 803 ms
51,456 KB |
testcase_37 | AC | 523 ms
37,108 KB |
testcase_38 | AC | 722 ms
47,684 KB |
testcase_39 | AC | 1,183 ms
68,148 KB |
testcase_40 | AC | 502 ms
36,816 KB |
testcase_41 | AC | 412 ms
32,272 KB |
testcase_42 | AC | 1,331 ms
73,476 KB |
testcase_43 | AC | 790 ms
68,156 KB |
testcase_44 | AC | 304 ms
31,480 KB |
testcase_45 | AC | 459 ms
43,536 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[pp] = i dfs(n-1) ret = 1 for i in range(n): ret = ret * ans[i] % MOD print(ret)