Universal Cup Judging System

Universal Cup

Time Limit: 1.0 s Memory Limit: 1024 MB Total points: 100 Hackable ✓
Statistics

Gegeben ist ein unbewurzelter Baum mit $n$ Knoten, wobei jeder Knoten eine Farbe hat, die für den $i$-ten Knoten durch eine positive ganze Zahl $c_i$ dargestellt wird.

Angenommen, $S$ und $T$ sind Mengen von Knoten im Baum. Ein geordnetes Paar $(S, T)$ gilt als gültig, wenn alle folgenden Bedingungen erfüllt sind:

  • Sowohl $S$ als auch $T$ sind nicht-leere Mengen;
  • Alle Knoten in $S$ haben dieselbe Farbe, und alle Knoten in $T$ haben dieselbe Farbe;
  • Der Teilgraph $\text{sub}(S)$ der Knotenmenge $S$ ist definiert als der kleinste zusammenhängende Teilgraph des Baums, der die Knotenmenge $S$ enthält, und analog für $\text{sub}(T)$. Es wird gefordert, dass sich $\text{sub}(S)$ und $\text{sub}(T)$ nicht schneiden, d. h. sie haben keine gemeinsamen Knoten oder Kanten.

Bestimme die Anzahl aller gültigen geordneten Paare $(S, T)$ modulo $998244353$. Es ist zu beachten, dass für $S \neq T$ das geordnete Paar $(S, T)$ als verschieden vom geordneten Paar $(T, S)$ betrachtet wird.

Eingabe

Dieses Problem enthält mehrere Testfälle. Die erste Zeile der Eingabe enthält eine ganze Zahl $T$ ($1 \le T \le 10^4$), die die Anzahl der Testfälle angibt.

Für jeden Testfall:

Die erste Zeile enthält eine ganze Zahl $n$ ($1 \le n \le 2 \times 10^5$), die die Anzahl der Knoten im unbewurzelten Baum angibt.

Die zweite Zeile enthält $n$ ganze Zahlen $c_1, c_2, \dots, c_n$ ($1 \le c_i \le n$), die die Farbe jedes Knotens angeben.

Die nächsten $n - 1$ Zeilen enthalten jeweils zwei ganze Zahlen $u, v$ ($1 \le u, v \le n, u \neq v$), die die Kanten des Baums darstellen. Es ist garantiert, dass diese $n - 1$ Kanten eine Baumstruktur bilden.

Die Daten garantieren, dass die Gesamtsumme von $n$ über alle Testfälle hinweg $2 \times 10^5$ nicht überschreitet.

Ausgabe

Gib für jeden Testfall eine einzelne Zeile mit einer ganzen Zahl aus, die die Anzahl der geordneten Paare modulo $998244353$ darstellt.

Beispiel

Beispieleingabe 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

Beispielausgabe 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.