#P14902. [ICPC 2018 Yokohama R] Colorful Tree

[ICPC 2018 Yokohama R] Colorful Tree

题目描述

A tree structure with some colors associated with its vertices and a sequence of commands on it are given. A command is either an update operation or a query on the tree. Each of the update operations changes the color of a specified vertex, without changing the tree structure. Each of the queries asks the number of edges in the minimum connected subgraph of the tree that contains all the vertices of the specified color.

Your task is to find answers of each of the queries, assuming that the commands are performed in the given order.

输入格式

The input consists of a single test case of the following format.

$$\begin{aligned} & n \\ & a_1 \; b_1 \\ & \vdots \\ & a_{n-1} \; b_{n-1} \\ & c_1 \; \ldots \; c_n \\ & m \\ & command_1 \\ & \vdots \\ & command_m \end{aligned}$$

The first line contains an integer nn (2n1000002 \leq n \leq 100\,000), the number of vertices of the tree. The vertices are numbered 11 through nn. Each of the following n1n-1 lines contains two integers aia_i (1ain1 \leq a_i \leq n) and bib_i (1bin1 \leq b_i \leq n), meaning that the ii-th edge connects vertices aia_i and bib_i. It is ensured that all the vertices are connected, that is, the given graph is a tree. The next line contains nn integers, c1c_1 through cnc_n, where cjc_j (1cj1000001 \leq c_j \leq 100\,000) is the initial color of vertex jj. The next line contains an integer mm (1m1000001 \leq m \leq 100\,000), which indicates the number of commands. Each of the following mm lines contains a command in the following format.

U  xk  yk\text{U} \; x_k \; y_k

or

Q  yk\text{Q} \; y_k

When the kk-th command starts with U, it means an update operation changing the color of vertex xkx_k (1xkn1 \leq x_k \leq n) to yky_k (1yk1000001 \leq y_k \leq 100\,000). When the kk-th command starts with Q, it means a query asking the number of edges in the minimum connected subgraph of the tree that contains all the vertices of color yky_k (1yk1000001 \leq y_k \leq 100\,000).

输出格式

For each query, output the number of edges in the minimum connected subgraph of the tree containing all the vertices of the specified color. If the tree doesn’t contain any vertex of the specified color, output 1-1 instead.

5
1 2
2 3
3 4
2 5
1 2 1 2 3
11
Q 1
Q 2
Q 3
Q 4
U 5 1
Q 1
U 3 2
Q 1
Q 2
U 5 4
Q 1
2
2
0
-1
3
2
2
0