Universal Cup Judging System

Universal Cup

时间限制: 1.0 s 内存限制: 1024 MB 总分: 100 可 Hack ✓
统计

Se da un árbol sin raíz que contiene $n$ nodos, donde cada nodo tiene un color representado por un entero positivo $c_i$ para el $i$-ésimo nodo.

Supongamos que $S$ y $T$ son conjuntos de nodos en el árbol. Un par ordenado $(S, T)$ se considera válido si se cumplen todas las condiciones siguientes:

  • Tanto $S$ como $T$ son conjuntos no vacíos;
  • Todos los nodos en $S$ tienen el mismo color, y todos los nodos en $T$ tienen el mismo color;
  • El subgrafo $\text{sub}(S)$ del conjunto de nodos $S$ se define como el subgrafo conexo más pequeño del árbol que contiene al conjunto de nodos $S$, y de manera similar para $\text{sub}(T)$. Se requiere que $\text{sub}(S)$ y $\text{sub}(T)$ no se intersequen, lo que significa que no tienen nodos ni aristas en común.

Determina el número de todos los pares ordenados válidos $(S, T)$ módulo $998244353$. Cabe señalar que si $S \neq T$, entonces el par ordenado $(S, T)$ se considera diferente del par ordenado $(T, S)$.

Entrada

Este problema contiene múltiples casos de prueba. La primera línea de la entrada contiene un entero $T$ ($1 \le T \le 10^4$), que representa el número de casos de prueba.

Para cada caso de prueba:

La primera línea contiene un entero $n$ ($1 \le n \le 2 \times 10^5$), que representa el número de nodos en el árbol sin raíz.

La segunda línea contiene $n$ enteros $c_1, c_2, \dots, c_n$ ($1 \le c_i \le n$), que representan el color de cada nodo.

Las siguientes $n - 1$ líneas contienen cada una dos enteros $u, v$ ($1 \le u, v \le n, u \neq v$), que representan las aristas del árbol. Se garantiza que estas $n - 1$ aristas forman una estructura de árbol.

Los datos garantizan que la suma total de $n$ en todos los casos de prueba no supera $2 \times 10^5$.

Salida

Para cada caso de prueba, imprime una sola línea que contenga un entero, representando el número de pares ordenados módulo $998244353$.

Ejemplo

Entrada de ejemplo 1

3
2
1 1
1 2
5
1 1 1 2 2
1 2
2 3
3 4
4 5
5
1 2 1 2 1
1 2
1 3
2 4
2 5

Salida de ejemplo 1

2
54
42

Discussions

About Discussions

The discussion section is only for posting: General Discussions (problem-solving strategies, alternative approaches), and Off-topic conversations.

This is NOT for reporting issues! If you want to report bugs or errors, please use the Issues section below.

Open Discussions 0
No discussions in this category.

Issues

About Issues

If you find any issues with the problem (statement, scoring, time/memory limits, test cases, etc.), you may submit an issue here. A problem moderator will review your issue.

Guidelines:

  1. This is not a place to publish discussions, editorials, or requests to debug your code. Issues are only visible to you and problem moderators.
  2. Do not submit duplicated issues.
  3. Issues must be filed in English or Chinese only.
Active Issues 0
No issues in this category.
Closed/Resolved Issues 0
No issues in this category.