Wannafly挑战赛1 A Treepath

最新推荐文章于 2020-03-22 11:22:17 发布

qq_34287501

最新推荐文章于 2020-03-22 11:22:17 发布

阅读量393

点赞数

CC 4.0 BY-SA版权

分类专栏：牛客网文章标签：树搜索

本文链接：https://blog.youkuaiyun.com/qq_34287501/article/details/78233350

牛客网专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一道关于在树形结构中寻找偶数长度路径数量的问题，并提供了一种高效的求解算法实现。通过DFS深度优先搜索策略遍历树节点，统计路径长度为偶数的路径总数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：https://www.nowcoder.com/acm/contest/15/A

时间限制：1秒空间限制：32768K

题目描述

给定一棵n个点的树，问其中有多少条长度为偶数的路径。路径的长度为经过的边的条数。x到y与y到x被视为同一条路径。路径的起点与终点不能相同。

输入描述:

第一行一个数n表示点的个数；
接下来n-1行，每行两个整数x，y表示边；
保证输入数据形成一棵树；
1<=n<=100000

输出描述:

一行一个整数表示答案。

示例1

输入

3
1 2
1 3

输出

解析：它是一棵树，不是一个图，随便找一个顶点搜一下，搜完之后必然把所有顶点都覆盖了，距离为偶数的只能连接偶数，奇数只能连接奇数，就是n*(n-1)/2

代码：

#include<bits/stdc++.h>
using namespace std;
 
vector<int> mp[100009];
int used[100009];
int res[100009];
 
void dfs(int u, int f)
{
    used[u] = 1;
    res[u] = f;
    for(int j = 0; j < mp[u].size(); j++)
    {
        if(used[mp[u][j]]) continue;
        dfs(mp[u][j], !f);
    }
 
}
 
 
int main()
{
    int n, x, y;
    scanf("%d", &n);
    for(int i = 1; i < n; i++)
    {
        scanf("%d%d", &x, &y);
        mp[x].push_back(y);
        mp[y].push_back(x);
    }
    dfs(1, 0);
    long long o, e; o = e = 0;
    for(int i = 1; i <= n; i++)
        if(res[i]) o++;
        else e++;
 
    printf("%lld\n", o*(o-1)/2+e*(e-1)/2);
    return 0;
}