问题 D: 涂色树

内存限制:1024 MB 时间限制:2.000 S
评测方式:文本比较 命题人:
提交:18 解决:4

题目描述

给定一棵包含 $N$ 个顶点的树,每个顶点都具有一个颜色。
我们定义一个好的顶点,即从根顶点 $1$ 到该顶点的最短路径中不包含与该顶点相同颜色的顶点(除了该顶点本身)。
现在需要找出树上所有满足这一条件的好的顶点。

输入

输入共 $N+1$ 行,
第 $1$ 行包括一个正整数 $N(2 \leq N \leq 10^5)$ ,
第 $2$ 行包括 $N$ 个空格分隔的正整数 $col_i(1 \leq col_i \leq 10^5)$ 代表第 $i$ 个顶点的颜色编号,
第 $3$ 到 $N+1$ 行每行包括两个正整数 $u,v(1 \leq u,v \leq N)$ ,表示树上存在一条边连接了点 $u,v$

输出

输出将所有好的顶点按升序打印,使用换行符作为分隔符。

样例输入 复制

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

样例输出 复制

1
2
3
4
6