結果
問題 | No.901 K-ary εxtrεεmε |
ユーザー | chineristAC |
提出日時 | 2020-09-27 22:41:16 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 807 ms / 3,000 ms |
コード長 | 2,091 bytes |
コンパイル時間 | 401 ms |
コンパイル使用メモリ | 87,200 KB |
実行使用メモリ | 210,968 KB |
最終ジャッジ日時 | 2023-09-13 03:51:31 |
合計ジャッジ時間 | 21,733 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 483 ms
210,968 KB |
testcase_01 | AC | 89 ms
71,800 KB |
testcase_02 | AC | 106 ms
77,740 KB |
testcase_03 | AC | 105 ms
77,820 KB |
testcase_04 | AC | 106 ms
77,624 KB |
testcase_05 | AC | 108 ms
77,696 KB |
testcase_06 | AC | 107 ms
77,852 KB |
testcase_07 | AC | 756 ms
175,428 KB |
testcase_08 | AC | 749 ms
176,472 KB |
testcase_09 | AC | 740 ms
168,564 KB |
testcase_10 | AC | 769 ms
175,440 KB |
testcase_11 | AC | 766 ms
177,272 KB |
testcase_12 | AC | 712 ms
166,956 KB |
testcase_13 | AC | 757 ms
169,320 KB |
testcase_14 | AC | 729 ms
164,800 KB |
testcase_15 | AC | 715 ms
165,296 KB |
testcase_16 | AC | 719 ms
164,680 KB |
testcase_17 | AC | 782 ms
179,384 KB |
testcase_18 | AC | 795 ms
182,588 KB |
testcase_19 | AC | 807 ms
180,820 KB |
testcase_20 | AC | 789 ms
179,372 KB |
testcase_21 | AC | 782 ms
180,956 KB |
testcase_22 | AC | 692 ms
160,112 KB |
testcase_23 | AC | 703 ms
163,724 KB |
testcase_24 | AC | 687 ms
164,044 KB |
testcase_25 | AC | 668 ms
161,780 KB |
testcase_26 | AC | 699 ms
167,248 KB |
testcase_27 | AC | 769 ms
185,244 KB |
testcase_28 | AC | 792 ms
186,000 KB |
testcase_29 | AC | 797 ms
186,092 KB |
ソースコード
import sys from collections import deque input = sys.stdin.readline sys.setrecursionlimit(2*10**5) N = int(input()) edge = [[] for i in range(N)] for i in range(N-1): u,v,w = map(int,input().split()) edge[u].append((v,w)) edge[v].append((u,w)) color_tree_edge = [set([]) for i in range(N)] Q = int(input()) color = [list(map(int,input().split()))[1:] for i in range(Q)] for i in range(Q): for v in color[i]: color_tree_edge[v].add(i) #合流点の検出 deq = deque([0]) parent = [-1 for i in range(N)] depth = [0 for i in range(N)] parent[0] = 0 topo = [] while deq: v = deq.popleft() topo.append(v) for nv,cost in edge[v]: if parent[nv]==-1: parent[nv] = v depth[nv] = depth[v] + cost deq.append(nv) topo = topo[::-1] node = [set([]) for v in range(N)] size = [1 for i in range(N)] for i in range(Q): for v in color[i]: node[v].add(i) for v in topo: for nv,cost in edge[v]: if nv!=parent[v]: size[v] += size[nv] if len(node[v]) < len(node[nv]): node[nv],node[v] = node[v],node[nv] for nv,cost in edge[v]: if nv!=parent[v]: for color in node[nv]: if color in node[v]: color_tree_edge[v].add(color) else: node[v].add(color) node[nv] = None res = [0 for i in range(Q)] #辺の張り直し vertex = [] def euler_tour(v,pv): vertex.append(v) for nv,cost in edge[v]: if nv!=pv: euler_tour(nv,v) vertex.append(v) euler_tour(0,0) color_parent = [-1 for i in range(N+1)] visit = [None for i in range(N+1)] for v in vertex: if visit[v] is None: visit[v] = {} for nc in color_tree_edge[v]: if color_parent[nc]!=-1: res[nc] += depth[v] - depth[color_parent[nc]] visit[v][nc] = color_parent[nc] color_parent[nc] = v else: for nc in visit[v]: color_parent[nc] = visit[v][nc] for i in range(Q): print(res[i])