牛客-ZJH and Monkeys（树的dfs）

原题链接

题意：给定一棵有n个节点树，你需要从树中选3个节点（可以重复），如果满足abs(abs(c - a) - abs(c - b))的奇偶性与n的相同，则答案加1。

解题思路：首先，我们需要找到每个节点到1的距离，利用他们的相对距离来判断两个点的距离之差来的奇偶性，所以我们可以统计每个点到1节点是奇数还是偶数。其次我们需要分情况讨论n的奇偶性

1.n为奇数则a，b，c可以分别为奇偶偶，偶奇奇，偶奇偶，奇偶奇

2.n为偶数则a，b，c可以分别为偶偶偶，奇奇奇，奇奇偶，偶偶奇

C++代码

#include <bits/stdc++.h>
#define int long long 
using namespace std;
const int N = 1e5 + 10, M = 2 * N;
int h[N], e[M], w[M], ne[M], idx;
int odd, even;
int n;

void add(int a, int b, int c)  // 添加一条边a->b，边权为c
{
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++ ;
}


void dfs(int u, int fa, int dist)
{
    if (dist & 1) even ++ ;
    else odd ++ ;
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (j == fa) continue;
        dfs(j, u, dist + 1);
    }
}

signed main()
{
    int t;
    cin >> t;
    while (t -- )
    {
        idx = 0;
        memset(h, -1, sizeof h);
        odd = even = 0;
        cin >> n;
        int a, b;
        for (int i = 1; i < n; i ++ )
        {
            cin >> a >> b;
            add(a, b, 1), add(b, a, 1);
        }
        dfs(1, 0, 0);
        if (n & 1) cout << 2 * (odd * even * even + even * odd * odd) << endl;
        else cout << (even * even + odd * odd) * n << endl;
    }
    return 0;
}