藍橋杯2023年A組-試題E-顏色平衡樹

DawnTraveler發表於2024-04-13

0.題目:顏色平衡樹

題目描述

給定一棵樹,結點由 \(1\)\(n\) 編號,其中結點 \(1\) 是樹根。樹的每個點有一個顏色 \(C_i\)

如果一棵樹中存在的每種顏色的結點個數都相同,則我們稱它是一棵顏色平衡樹。

求出這棵樹中有多少個子樹是顏色平衡樹。

輸入格式

輸入的第一行包含一個整數 \(n\),表示樹的結點數。

接下來 \(n\) 行,每行包含兩個整數 \(C_i,F_i\),用一個空格分隔,表示第 \(i\) 個結點的顏色和父親結點編號。

特別地,輸入資料保證 \(F_1\)\(0\),也即 \(1\) 號點沒有父親結點。保證輸入資料是一棵樹。

輸出格式

輸出一行包含一個整數表示答案。

樣例 #1

樣例輸入 #1

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

樣例輸出 #1

4

提示

【樣例說明】

編號為 \(1,3,5,6\)\(4\) 個結點對應的子樹為顏色平衡樹。

【評測用例規模與約定】

對於 \(30 \%\) 的評測用例,\(n \leq 200\)\(C_i \leq 200\)

對於 \(60 \%\) 的評測用例,\(n \leq 5000\)\(C_i \leq 5000\)

對於所有評測用例,\(1 \leq n \leq 200000\)\(1 \leq C_i \leq 200000\)\(0 \leq F_i<i\)

1.題解

相關文章