链接:https://ac.nowcoder.com/acm/contest/11221/E
来源:牛客网
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
幽怨火,憎恨焰,变异蛮牛续执念。
给定一棵根为 11,且是黑点的有根树。
每个白点相邻所有的点都是黑点,每个黑点相邻所有的点都是白点。换句话说,你可以从根结点开始,按照深度对每个点黑白染色。
现在对于一条两个端点分别是 u,vu,v 的链,定义其长度为:包含的黑点个数 -− 包含的白点个数。
请你数一数 长度最大 的链的个数。
输入描述:
全文第一行是 T(1\le T\le10^5)T(1≤T≤10
5
),表示数据组数;
接下来 TT 组数据,先输入一行一个正整数表示树的大小 n(1\le n\le2\times10^5,\sum n\le3\times10^6)n(1≤n≤2×10
5
,∑n≤3×10
6
);
接下来输入 n-1n−1 行每行两个正整数 u,v(1\le u,v\le n)u,v(1≤u,v≤n) 表示树的一条边。
输出描述:
输出 TT 行,每行一个整数表示答案。
示例1
输入
复制
1
6
1 2
2 3
3 4
4 5
5 6
输出
复制
6
说明
合法的链分别是:{1},{3},{5},{1,2,3},{3,4,5},{1,2,3,4,5}{1},{3},{5},{1,2,3},{3,4,5},{1,2,3,4,5}。
思路 :
- 显然这棵树第一层全黑,第二层全白,第三层全黑…因此,黑 - 白数量最多为1,要么就选一个黑点,要么就选起始皆为黑点,假设黑点个数为cnt,则答案为 c n t + C c n t 2 cnt + C^2_{cnt} cnt+Ccnt2
- 值得注意的是,最后输出公式中加法的两个部分都要先乘上
1ll
,否则会wa - 优化:每次清空边动态数组时,只要清理本次样例的范围
#include <iostream>
#include <vector>
using namespace std;
const int N = 2e5 + 10;
vector<int> e[N];
int cnt;
void dfs(int u, int fa, int col)
{
if (col) cnt ++ ;
for (auto v : e[u])
if (v != fa)
dfs(v, u, col ^ 1);
}
int main()
{
int _; cin >> _;
while (_ -- )
{
int n; scanf("%d", &n);
cnt = 0;
for (int i = 1; i <= n; i ++ ) e[i].clear();
for (int i = 0, u ,v; i < n - 1 && scanf("%d%d", &u, &v); i ++ )
e[u].push_back(v), e[v].push_back(u);
dfs(1, 0, 1);
printf("%lld\n", 1ll * cnt + 1ll * cnt * (cnt - 1) / 2);
}
}