挑战赛1 A Treepath

Treepath

时间限制：1秒 空间限制：32768K

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

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+100;

int n;
vector<int > g[N];
ll a[N][3];
ll ans;

void dfs(int u,int fa){
    for(int i=0;i<g[u].size();i++){
        int v=g[u][i];
        if(v==fa) continue ;
        dfs(v,u);
        a[u][0]+=a[v][1];
        a[u][1]+=a[v][0]+1;
    }
    for(int i=0;i<g[u].size();i++){
        int v=g[u][i];
        if(v==fa) continue;
        ll x=a[u][0]-a[v][1];
        ll y=a[u][1]-(a[v][0]+1);
        ans+=x*a[v][1]+y*(a[v][0]+1);
    }
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(1,0);
    ans/=2;
    for(int i=1;i<=n;i++)
    ans+=a[i][0];
    cout<<ans<<endl;
}