[蓝桥杯 2023 省 A] 颜色平衡树

Luogu
IDLGP9233
Time1000ms
Memory256MB
DifficultyP5
树上启发式合并2023蓝桥杯省赛
给定一棵树,结点由 $1$ 至 $n$ 编号,其中结点 $1$ 是树根。树的每个点有一个颜色 $C_i$。 如果一棵树中存在的每种颜色的结点个数都相同,则我们称它是一棵颜色平衡树。 求出这棵树中有多少个子树是颜色平衡树。 ## Input 输入的第一行包含一个整数 $n$,表示树的结点数。 接下来 $n$ 行,每行包含两个整数 $C_i,F_i$,用一个空格分隔,表示第 $i$ 个结点的颜色和父亲结点编号。 特别地,输入数据保证 $F_1$ 为 $0$,也即 $1$ 号点没有父亲结点。保证输入数据是一棵树。 ## Output 输出一行包含一个整数表示答案。 [samples] ## Note #### 【样例说明】 编号为 $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 2\times 10 ^ 5$,$1 \leq C_i \leq 2\times 10 ^ 5$,$0 \leq F_i<i$。
Samples
Input #1
6
2 0
2 1
1 2
3 3
3 4
1 4
Output #1
4
API Response (JSON)
{
  "problem": {
    "name": "[蓝桥杯 2023 省 A] 颜色平衡树",
    "description": {
      "content": "给定一棵树,结点由 $1$ 至 $n$ 编号,其中结点 $1$ 是树根。树的每个点有一个颜色 $C_i$。 如果一棵树中存在的每种颜色的结点个数都相同,则我们称它是一棵颜色平衡树。 求出这棵树中有多少个子树是颜色平衡树。",
      "description_type": "Markdown"
    },
    "platform": "Luogu",
    "limit": {
      "time_limit": 1000,
      "memory_limit": 262144
    },
    "difficulty": {
      "LuoguStyle": "P5"
    },
    "is_remote": true,
    "is_sync": true,
    "sync_url": null,
    "sign": "LGP9233"
  },
  "statements": [
    {
      "statement_type": "Markdown",
      "content": "给定一棵树,结点由 $1$ 至 $n$ 编号,其中结点 $1$ 是树根。树的每个点有一个颜色 $C_i$。\n\n如果一棵树中存在的每种颜色的结点个数都相同,则我们称它是一棵颜色平衡树。\n\n求出这棵树中有多少个子树是颜色平衡树。\n\n## Input\n\n输入的第一行包含一个整数 $n$,表示树的结点数。\n\n接下来 $n$ 行,每行包含两个整数 $C_i,F_i$,用一个空格分隔,表示第 $i$ 个结点的...",
      "is_translate": false,
      "language": "English"
    }
  ]
}
Full JSON Raw Segments