给定一棵以 1 为根的树,每个据点有整数标签 ti。对每个据点 i,要求统计在从据点 1 到该据点 i 的简单路径上,有多少无序对 (x,y)(x=y)满足标签相同。
核心做法:根到当前据点的前缀统计 + DFS(栈模拟)
freq,以及“当前路径上的相等标签对总数” pairs。在一个由 n 个据点构成的网络中,任意两个据点之间都由唯一的一条简单路径相连。据点编号为 1 到 n,第 i 个据点带有一个整数标签 ti。
对于每个 i(1≤i≤n),你需要计算:在从据点 1 到据点 i 的简单路径上,有多少对不同的据点具有相同的标签。换言之,统计这条路径上标签相同的无序点对的数量。
据点数量 n 满足 2≤n≤2×105,标签值 ti 满足 1≤ti≤109。保证网络连通且无环。
第一行包含一个整数 n,表示据点数量。
Scan the QR code below with WeChat to sign in
First-time scan will create your account automatically
请使用微信扫描下方二维码完成注册