问题 F: 树上 MEX 问题

内存限制:256 MB 时间限制:6 S
题面:Markdown 评测方式:文本比较 上传者:
提交:2 通过:2

题目描述

定义一张图的 MEX 为:在图中所有点的点权所构成的集合中,最小的没有出现过的非负整数。 给定一棵 $n$ 个点的树,点编号为 $1$ 到 $n$,第 $i$ 个点有点权 $a_i$,保证每个点的点权互不相同。求这棵树的所有[b]连通导出子图[/b]的 MEX 之和,答案对 $998244353$ 取模。 一个图的导出子图是指,由该图顶点的一个子集和该图中两端均在该子集的所有边的集合组成的图。

输入格式

第一行一个正整数 $T$($1\le T\le 10^5$),表示测试数据组数。 对于每组数据,第一行包含一个整数 $n$($1\le n\le 10^5$),表示树的点数。 第二行包含 $n$ 个整数 $a_1,a_2,\ldots ,a_n$($0\le a_i < n$),表示每个点的点权。保证每个点的点权互不相同。 接下来 $n-1$ 行描述了这棵树,每行包含两个整数 $u,v$($1\le u,v\le n$),表示点 $u$ 和点 $v$ 之间有一条边。保证输入是一棵树。 保证对于所有测试数据,$n$ 之和不超过 $10^6$。

输出格式

对于每组测试数据,输出一行一个整数,表示答案对 $998244353$ 取模后的结果。

输入样例 复制

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

输出样例 复制

40
24
18