Wannafly挑战赛1A-(树形DP）

题解:从叶子结点到根计算一下每个子结点的从该结点开始计算深度是偶数深度是奇数的个数即可,然后顺便求一下这个点下面深度是奇数的总个数和偶数的总个数即可

然后方案数就是每个点的下面点奇数个数乘上奇数个数，偶数个数乘偶数个数

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long int ll;
#define x first
#define y second
const int mx = 1e5+5;
int n;
vector<int>g[mx];
pair<int,int> dp[mx];
int main(){
    while(scanf("%d",&n)!=EOF){
        for(int i = 1; i <= n; i++)
            g[i].clear();
        for(int i = 1; i < n; i++){
            int u,v;
            scanf("%d%d",&u,&v);
            if(u>v)
                swap(u,v);
            g[u].push_back(v);
        }
        long long ans = 0;
        for(int u = n; u >= 1; u--){
            dp[u].x = 1;
            dp[u].y = 0;
            for(auto v : g[u]){
                ans += 1ll*dp[u].x*dp[v].y;
                ans += 1ll*dp[u].y*dp[v].x;
                dp[u].y += dp[v].x;
                dp[u].x += dp[v].y;
            }
        }
        printf("%lld\n",ans);
    }
}