Treepath

Treepath

题目描述

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

输入描述:

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

输出描述:

一行一个整数表示答案。

示例1

输入

3
1 2
1 3

输出

1

dfs，假设1为最顶层，遍历每一个节点，求出奇数层和偶数层的数量。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 100010;
vector<int> vs[N];
bool vis[N];
ll ans1, ans2;
void dfs(int x, int de) {
    vis[x] = true;
    for(int i = 0; i < vs[x].size(); i ++) {
        int v = vs[x][i];
        if(!vis[v]) {
            if((de+1) %2) ans1 ++;
            else ans2 ++; 
            dfs(v, de+1);
        }
    }
}
void init(){
    ans1 = ans2 = 0;
    memset(vis, false, sizeof(vis));
    for(int i = 1; i < N; i ++) vs[i].clear();
}
int main() {
    int n, x, y;
    while(scanf("%d", &n)!=EOF) {
        init();
        for(int i = 1; i < n; i ++) {
            scanf("%d %d", &x, &y);
            vs[x].push_back(y);
            vs[y].push_back(x);
        }
        dfs(1, 1);
        ans1++;
        // printf("%d %d
", ans1, ans2);
        ll cnt = (ans1-1)*ans1 + (ans2-1)*ans2;
        printf("%lld
",cnt/2);        
    }    
    return 0;
}