SZ#G6DFS27. 【GESP强化 六级】根路径上的独特颜色

提交1 通过1
通过率100%
时间限制2000ms
内存限制256MiB

题目描述

珅泽教育的小婷老师正在准备一项搜索实践,她请小泽完成下面的任务。

一棵以顶点 11 为根的树有 NN 个顶点,顶点 ii 的颜色编号为 CiC_i。对顶点 xx,考虑从根 11xx 的唯一简单路径。

如果这条路径上没有其他顶点与 xx 的颜色相同,就称顶点 xx 是好顶点。请把所有好顶点的编号按从小到大的顺序输出。

输入格式

第一行输入 NN

第二行输入 NN 个颜色编号。

接下来 N1N-1 行输入树边。

输出格式

每行输出一个好顶点编号,按升序排列。

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

数据范围与约定

  • 1N1051 \le N \le 10^5
  • 1Ci1051 \le C_i \le 10^5