Codeforces Round #435 (Div. 2), problem: (B) Mahmoud and Ehab and the bipartiteness 【染色法dfs】

题意

给定一个n 个节点 n-1 条边的图，求最多还能加几条边，保证这个图不存在重边，自环，并且是一个二分图

思路

树必然是二分图，一定有n-1条边，我们先dfs对其二分染色，把点分为两个集合，两集合大小的积|s1|*|s2|就是最多能有的边数，减去原边数即是所求。

code

#include<bits/stdc++.h>
#define endl '\n'
using namespace std;
typedef long long ll;
int n;
const int maxn=1e5+5;
int vis[maxn];
vector<int> G[maxn];
ll n1,n2;
void dfs(int u,int t){
    vis[u]=1;
    if(t&1) n1++;
    else n2++;
    for(auto v:G[u]){
        if(!vis[v])
            dfs(v,t+1);
    }
}
int main(){
    cin>>n;
    n1=0,n2=0;
    for(int i=0;i<n-1;i++){
        int u,v;
        cin>>u>>v;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1,0);
    cout<<n1*n2-(n-1)<<endl;
    return 0;
}

学如逆水行舟，不进则退