[每日一题200414]Treepath

Treepath

https://ac.nowcoder.com/acm/problem/14248

传送门

题意

给定一棵n个点的树,问其中有多少条长度为偶数的路径。

tags

  • dfs/bfs
  • 思维

分析

以节点开始,记录每个点的深度。设奇偶深度的点个数分别为:,则ans=
具体证明过程如下图:
图片说明

参考代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 100000 + 10;
int head[maxn], cnt;
int dep[maxn];
struct node {
    int v, next;
} e[maxn << 1];

void add(int u, int v) {
    e[cnt].v = v;
    e[cnt].next = head[u];
    head[u] = cnt++;
}

void init() {
    memset(head, -1, sizeof(head));
    cnt = 0;
}

void dfs(int u, int v) {
    dep[u] = dep[v] + 1;
    for (int i = head[u]; ~i; i = e[i].next) {
        int x = e[i].v;
        if (v == x) continue;
        dfs(x, u);
    }
}

int main(void) {
    int n;
    init();
    cin >> n;
    for (int i = 1; i < n; i++) {
        int u, v;
        cin >> u >> v;
        add(u, v);
        add(v, u);
    }
    dfs(1, 1);
    ll x = 0, y = 0;
    for (int i = 1; i <= n; i++) {
        if (dep[i] & 1) {
            x++;
        } else {
            y++;
        }
    }
    cout << (x - 1) * x / 2 + (y - 1) * y / 2 << endl;

    return 0;
}
全部评论

相关推荐

不愿透露姓名的神秘牛友
昨天 18:28
点赞 评论 收藏
分享
有担当的灰太狼又在摸鱼:零帧起手查看图片
点赞 评论 收藏
分享
05-12 11:09
已编辑
门头沟学院 后端
已注销:没必要放这么多专业技能的描述。这些应该是默认已会的,写这么多行感觉在凑内容。项目这块感觉再包装包装吧,换个名字,虽然大家的项目基本都是网上套壳的,但是你这也太明显了。放一个业务项目,再放一个技术项目。技术项目,例如中间件的一些扩展和尝试。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务