Wannafly挑战赛1 A.Treepath(dfs)

题目:https://www.nowcoder.com/acm/contest/15/A
题目描述
给定一棵n个点的树,问其中有多少条长度为偶数的路径。路径的长度为经过的边的条数。x到y与y到x被视为同一条路径。路径的起点与终点不能相同。
输入描述:
第一行一个数n表示点的个数;
接下来n-1行,每行两个整数x,y表示边;
保证输入数据形成一棵树;
1<=n<=100000
输出描述:
一行一个整数表示答案。
示例1
输入

3
1 2
1 3
输出

1
思路:dfs记录奇偶个数,O(n)
代码:

#include<bits/stdc++.h>
using namespace std;

const int N = 100005;
int num[2],vis[N];
vector<int> G[N];

void dfs(int u,int cnt)
{
    num[cnt]++;
    vis[u] = 1;
    for(auto v : G[u])
    {
        if(vis[v])
            continue;
        dfs(v,1-cnt);
    }
}
int main()
{
    int n,u,v;
    scanf("%d",&n);
    for(int i = 0;i < n-1;i++)
    {
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1,0);
    long long sum = num[0]*(num[0]-1ll)/2 + num[1]*(num[1]-1ll)/2;
    printf("%lld\n",sum);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值