#S1587. 采花的小女孩

    ID: 24 传统题 文件IO:flower 5000ms 1024MiB 尝试: 0 已通过: 0 难度: (无) 上传者: 标签>搜索DFS数据结构主席树树结构树链剖分DFS序整体二分G11295

采花的小女孩

题目描述

在花园中有 nn 个花圃,在花圃 ii,共有 bib_i 朵花,这些花属于种类 sis_i

此外,还有 n1n-1 条小径,每条小径连接两花圃 uju_jvjv_j(双向)。并且满足小女孩从每一个花圃出发都能到达其它所有花圃。

贾文景邀请了 dd 个小女孩,并计划依次让她们从花圃 11 开始采花。每个小女孩有一个目标花圃 hk1h_k \neq 1,她们会沿着最短路径到目标花圃,并在经过的每个花圃中采一朵花(包括花圃 11)。

每次采花会减少花圃中 11 朵花(如果花的数量不为 00)。请帮助贾文景计算每个小女孩过程中采到的花有多少种不同的种类。

输入格式

  • 第一行包含两个整数 nndd,分别表示花圃数量和小女孩数量。

  • 第二行包含 nn 个整数 b1,b2,,bnb_1, b_2, \dots, b_n,表示每个花圃中花的数量。

  • 第三行包含 nn 个整数 s1,s2,,sns_1, s_2, \dots, s_n,表示每个花圃花的种类。

  • 第四行包含 dd 个整数 h1,h2,,hdh_1, h_2, \dots, h_d,表示每个小女孩的目标花圃。

  • 接下来 n1n-1 行,每行包含两个整数 uju_jvjv_j,表示一条连接花圃 uju_jvjv_j 的小径。

输出格式

输出一行包含 dd 个整数,第 kk 个整数表示第 kk 个小女孩采到的不同种类的数量。

样例 #1

5 6
4 1 0 3 1
1 3 2 2 1
2 5 4 3 4 2
5 2
2 1
1 4
1 3
2 1 2 1 1 0

样例 #2

7 4
0 2 4 4 0 1 3
6 1 6 2 2 2 1
7 5 2 4
4 1
4 5
6 2
1 6
1 3
6 7

输出 #2

2 1 1 1

说明/提示

【样例解释】

对于样例 11,第一个小女孩走到花圃 22,采到 11 朵种类 11 的花和 11 朵种类 22 的花。第二个小女孩走到花圃 55,采到种类 11 的花,总共采到 11 个种类。

对于样例 22,每个小女孩采到的不同种类分别是 2,1,1,12, 1, 1, 1

【数据范围】

  • 2n2×1052 \leq n \leq 2 \times 10^5
  • 1d2×1061 \leq d \leq 2 \times 10^6
  • 1sin1 \leq s_i \leq n0bid0 \leq b_i \leq d1uj,vj<n1 \leq u_j, v_j < n
子任务编号 分值 额外限制条件
11 1010 n,d1000n, d \leq 1000
22 bi=db_i = d
33 1212 bi10b_i \leq 10
44 uj=ju_j = j, vj=j+1v_j = j + 1
55 3737 si=is_i = i
66 1616 d2×105d \leq 2 \times 10^5
77 33 无额外限制

大数据