65035 - [NOIP2018 提高组] 保卫王国 defense
Z 国有 n 座城市,(n - 1) 条双向道路,每条双向道路连接两座城市,且任意两座城市都能通过若干条道路相互到达。
Z 国的国防部长小 Z 要在城市中驻扎军队。驻扎军队需要满足如下几个条件:
- 一座城市可以驻扎一支军队,也可以不驻扎军队。
- 由道路直接连接的两座城市中至少要有一座城市驻扎军队。
- 在城市里驻扎军队会产生花费,在编号为 i 的城市中驻扎军队的花费是 p_i。
小 Z 很快就规划出了一种驻扎军队的方案,使总花费最小。但是国王又给小 Z 提出了 m 个要求,每个要求规定了其中两座城市是否驻扎军队。小 Z 需要针对每个要求逐一给出回答。具体而言,如果国王提出的第 j 个要求能够满足上述驻扎条件(不需要考虑第 j 个要求之外的其它要求),则需要给出在此要求前提下驻扎军队的最小开销。如果国王提出的第 j 个要求无法满足,则需要输出 -1。现在请你来帮助小 Z。
输入
第一行有两个整数和一个字符串,依次表示城市数 n,要求数 m 和数据类型 type。type 是一个由大写字母 A
,B
或 C
和一个数字 1
,2
,3
组成的字符串。它可以帮助你获得部分分。你可能不需要用到这个参数。这个参数的含义在【数据规模与约定】中 有具体的描述。
第二行有 n 个整数,第 i 个整数表示编号 i 的城市中驻扎军队的花费 p_i。
接下来 (n - 1) 行,每行两个整数 u,v,表示有一条 u 到 v 的双向道路。
接下来 m 行,每行四个整数 a, x, b, y,表示一个要求是在城市 a 驻扎 x 支军队,在城市 b 驻扎 y 支军队。其中,x,y 的取值只有 0 或 1:
- 若 x 为 0,表示城市 a 不得驻扎军队。
- 若 x 为 1,表示城市 a 必须驻扎军队。
- 若 y 为 0,表示城市 b 不得驻扎军队。
- 若 y 为 1,表示城市 b 必须驻扎军队。
输入文件中每一行相邻的两个数据之间均用一个空格分隔。
输出
输出共 m 行,每行包含一个个整数,第 j 行表示在满足国王第 j 个要求时的最小开销, 如果无法满足国王的第 j 个要求,则该行输出 -1。
样例
输入
5 3 C3 2 4 1 3 9 1 5 5 2 5 3 3 4 1 0 3 0 2 1 3 1 1 0 5 0
输出
12 7 -1
提示
样例 1 解释
- 对于第一个要求,在 4 号和 5 号城市驻扎军队时开销最小。
- 对于第二个要求,在 1 号、2 号、3 号城市驻扎军队时开销最小。
- 第三个要求是无法满足的,因为在 1 号、5 号城市都不驻扎军队就意味着由道路直接连 接的两座城市中都没有驻扎军队。
数据规模与约定
测试点编号 | \text{type} | n = m= |
---|---|---|
1,2 | A3 | 10 |
3,4 | C3 | 10 |
5,6 | A3 | 100 |
7 | C3 | 100 |
8,9 | A3 | 2\times 10^3 |
10,11 | C3 | 2\times 10^3 |
12,13 | A1 | 10^5 |
14, 15, 16 | A2 | 10^5 |
17 | A3 | 10^5 |
18,19 | B1 | 10^5 |
20,21 | C1 | 10^5 |
22 | C2 | 10^5 |
23, 24, 25 | C3 | 10^5 |
数据类型的含义:
A
:城市i与城市i + 1直接相连。B
:任意城市与城市 1 的距离不超过 100(距离定义为最短路径上边的数量),即如果这 棵树以 1 号城市为根,深度不超过 100。C
:在树的形态上无特殊约束。1
:询问时保证a = 1,x = 1,即要求在城市 1 驻军。对b,y没有限制。2
:询问时保证a,b是相邻的(由一条道路直接连通)3
:在询问上无特殊约束。
对于 100\%的数据,保证 1 \leq n,m ≤ 10^5,1 ≤ p_i ≤ 10^5,1 \leq u, v, a, b \leq n,a \neq b,x, y \in {0, 1}。
来源
NOIP2018 提高组 D2T3