Universal Cup Judging System

Universal Cup

Límite de tiempo: 1.0 s Límite de memoria: 1024 MB Puntuación total: 100 Hackeable ✓
Estadísticas

On vous donne un arbre non enraciné contenant $n$ nœuds, où chaque nœud possède une couleur représentée par un entier strictement positif $c_i$ pour le $i$-ème nœud.

Supposons que $S$ et $T$ soient des ensembles de nœuds de l'arbre. Un couple $(S, T)$ est considéré comme valide si toutes les conditions suivantes sont réunies :

  • $S$ et $T$ sont tous deux des ensembles non vides ;
  • Tous les nœuds de $S$ ont la même couleur, et tous les nœuds de $T$ ont la même couleur ;
  • Le sous-graphe $\text{sub}(S)$ de l'ensemble de nœuds $S$ est défini comme le plus petit sous-graphe connexe de l'arbre qui contient l'ensemble de nœuds $S$, et de même pour $\text{sub}(T)$. On exige que $\text{sub}(S)$ et $\text{sub}(T)$ ne s'intersectent pas, c'est-à-dire qu'ils n'aient aucun nœud ni aucune arête en commun.

Déterminez le nombre de tous les couples valides $(S, T)$ modulo $998244353$. Notez que si $S \neq T$, alors le couple $(S, T)$ est considéré comme différent du couple $(T, S)$.

Entrée

Ce problème contient plusieurs jeux de tests. La première ligne de l'entrée contient un entier $T$ ($1 \le T \le 10^4$), représentant le nombre de jeux de tests.

Pour chaque jeu de tests :

La première ligne contient un entier $n$ ($1 \le n \le 2 \times 10^5$), représentant le nombre de nœuds dans l'arbre non enraciné.

La deuxième ligne contient $n$ entiers $c_1, c_2, \dots, c_n$ ($1 \le c_i \le n$), représentant la couleur de chaque nœud.

Les $n - 1$ lignes suivantes contiennent chacune deux entiers $u, v$ ($1 \le u, v \le n, u \neq v$), représentant les arêtes de l'arbre. Il est garanti que ces $n - 1$ arêtes forment une structure d'arbre.

Les données garantissent que la somme totale de $n$ sur l'ensemble des jeux de tests ne dépasse pas $2 \times 10^5$.

Sortie

Pour chaque jeu de tests, affichez une seule ligne contenant un entier, représentant le nombre de couples modulo $998244353$.

Exemple

Entrée d'exemple 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

Sortie d'exemple 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.