| 시간 제한 | 메모리 제한 | 제출 | 정답 | 맞힌 사람 | 정답 비율 |
|---|---|---|---|---|---|
| 2 초 | 1024 MB | 95 | 18 | 17 | 26.562% |
Consider a tree consisting of $N$ vertices, numbered from 0ドル$ to $N - 1$. Vertex 0ドル$ is called the root. Every vertex, except for the root, has a single parent. For every $i,ドル such that 1ドル ≤ i < N,ドル the parent of vertex $i$ is vertex $P[i],ドル where $P[i] < i$. We also assume $P[0] = -1$.
For any vertex $i$ (0ドル ≤ i < N$), the subtree of $i$ is the set of the following vertices:
The picture below shows an example tree consisting of $N = 6$ vertices. Each arrow connects a vertex to its parent, except for the root, which has no parent. The subtree of vertex 2ドル$ contains vertices 2ドル,ドル 3ドル,ドル 4ドル$ and 5ドル$. The subtree of vertex 0ドル$ contains all 6ドル$ vertices of the tree and the subtree of vertex 4ドル$ contains only vertex 4ドル$.
Each vertex is assigned a nonnegative integer weight. We denote the weight of vertex $i$ (0ドル ≤ i < N$) by $W[i]$.
Your task is to write a program that will answer $Q$ queries, each specified by a pair of positive integers $(L,R)$. The answer to the query should be computed as follows.
Consider assigning an integer, called a coefficient, to each vertex of the tree. Such an assignment is described by a sequence $C[0],\dots ,C[N - 1],ドル where $C[i]$ (0ドル ≤ i < N$) is the coefficient assigned to vertex $i$. Let us call this sequence a coefficient sequence. Note that the elements of the coefficient sequence can be negative, 0ドル,ドル or positive.
For a query $(L,R),ドル a coefficient sequence is called valid if, for every vertex $i$ (0ドル ≤ i < N$), the following condition holds: the sum of the coefficients of the vertices in the subtree of vertex $i$ is not less than $L$ and not greater than $R$.
For a given coefficient sequence $C[0],\dots ,C[N - 1],ドル the cost of a vertex $i$ is $|C[i]| \cdot W[i],ドル where $|C[i]|$ denotes the absolute value of $C[i]$. Finally, the total cost is the sum of the costs of all vertices. Your task is to compute, for each query, the minimum total cost that can be attained by some valid coefficient sequence.
It can be shown that for any query, at least one valid coefficient sequence exists.
You should implement the following two procedures:
void init(std::vector P, std::vector W)
long long query(int L, int R)
init in each test case.| 번호 | 배점 | 제한 |
|---|---|---|
| 1 | 10 | $Q ≤ 10$; $W[P[i]] ≤ W[i]$ for each $i$ such that 1ドル ≤ i < N$ |
| 2 | 13 | $Q ≤ 10$; $N ≤ 2,円 000$ |
| 3 | 18 | $Q ≤ 10$; $N ≤ 60,円 000$ |
| 4 | 7 | $W[i] = 1$ for each $i$ such that 0ドル ≤ i < N$ |
| 5 | 11 | $W[i] ≤ 1$ for each $i$ such that 0ドル ≤ i < N$ |
| 6 | 22 | $L = 1$ |
| 7 | 19 | No additional constraints. |
Consider the following calls:
init([-1, 0, 0], [1, 1, 1])
The tree consists of 3ドル$ vertices, the root and its 2ドル$ children. All vertices have weight 1ドル$.
query(1, 1)
In this query $L = R = 1,ドル which means the sum of coefficients in every subtree must be equal to 1ドル$. Consider the coefficient sequence $[-1, 1, 1]$. The tree and the corresponding coefficients (in shaded rectangles) are illustrated below.
For every vertex $i$ (0ドル ≤ i < 3$), the sum of the coefficients of all vertices in the subtree of $i$ is equal to 1ドル$. Hence, this coefficient sequence is valid. The total cost is computed as follows:
| Vertex | Weight | Coefficient | Cost |
|---|---|---|---|
| 0ドル$ | 1ドル$ | $-1$ | $|-1| \cdot 1 = 1$ |
| 1ドル$ | 1ドル$ | 1ドル$ | $|1| \cdot 1 = 1$ |
| 2ドル$ | 1ドル$ | 1ドル$ | $|1| \cdot 1 = 1$ |
Therefore the total cost is 3ドル$. This is the only valid coefficient sequence, therefore this call should return 3ドル$.
query(1, 2)
The minimum total cost for this query is 2ドル,ドル and is attained when the coefficient sequence is $[0, 1, 1]$.
Input format:
N P[1] P[2] ... P[N-1] W[0] W[1] ... W[N-2] W[N-1] Q L[0] R[0] L[1] R[1] ... L[Q-1] R[Q-1]
where $L[j]$ and $R[j]$ (for 0ドル ≤ j < Q$) are the input arguments in the $j$-th call to query. Note that the second line of the input contains only $N - 1$ integers, as the sample grader does not read the value of $P[0]$.
Output format:
A[0] A[1] ... A[Q-1]
where $A[j]$ (for 0ドル ≤ j < Q$) is the value returned by the $j$-th call to query.
Olympiad > International Olympiad in Informatics > IOI 2024 > Day 1 3번
C++17, C++20, C++17 (Clang), C++20 (Clang)