def solve(N, Q, edges, QS, ordered_edges):
from collections import defaultdict
color = [0] * N
uplink = [None] * N
root = 0
# graph to tree
children, parents = graph_to_tree(N, edges, root)
# ready lca
construct(N, children, root, parents)
def paint(start, end, c):
cur = start
while cur != end:
if color[cur] == 0:
color[cur] = c
uplink[cur] = end
cur = parents[cur]
else:
if query(uplink[cur], end) != end:
# uplink if avobe end
return
cur = uplink[cur]
for a, b, c in reversed(QS):
a -= 1
b -= 1
lca = query(a, b)
paint(a, lca, c)
paint(b, lca, c)
for frm, to in ordered_edges:
if parents[frm] == to:
print(color[frm])
else:
print(color[to])
頂点と辺の取り違え関連でこういうミスをしていた
uplinkで「過去に塗られた時のend」までジャンプした後、さらにparentに進んでる
頂点ぬりなら正しいのだが、辺ぬりなので正しくない
python
def paint(start, end, c):
cur = start
while cur != end:
if color[cur] == 0:
color[cur] = c
uplink[cur] = end
else:
if query(uplink[cur], end) != end:
# uplink if avobe end
return
cur = uplink[cur]
if cur == end: return
cur = parents[cur]