codeforces 862B B. Mahmoud and Ehab and the bipartiteness

http://codeforces.com/problemset/problem/862/B

题意：

给出一个有n个点的二分图和n-1条边，问现在最多可以添加多少条边使得这个图中不存在自环，重边，并且此图还是一个二分图。

思路：

想得比较复杂了。。。。其实既然已经给出了二分图并且有n-1条边，那么我们就一定可以用染色法对每一个点进行染色，从而将点划分为两个集合，然后从两个集合中分别任意选择一个点连边就行了。

一开始考虑二分图的基本属性是不存在奇数条边的环。。。并不需要这样，因为两个集合是分开的，从两个中分别任意选择一个连边，是肯定不会造成同一集合中的两点有边相连的。

代码：

#include <stdio.h>
#include <vector>
using namespace std;

vector<int> v[100005];
bool vis[100005];

int color[100005];

void dfs(int s)
{
    vis[s] = 1;

    if (color[s] == 0) color[s] = 1;

    for (int i = 0;i < v[s].size();i++)
    {
        int to = v[s][i];

        if (!vis[to])
        {
            if (color[s] == 1) color[to] = 2;
            else color[to] = 1;

            vis[to] = 1;
            dfs(to);
        }
    }
}

int main()
{
    int n;

    scanf("%d",&n);

    for (int i = 0;i < n - 1;i++)
    {
        int x,y;

        scanf("%d%d",&x,&y);

        v[x].push_back(y);
        v[y].push_back(x);
    }

    dfs(1);

    long long cnt1 = 0,cnt2 = 0;

    for (int i = 1;i <= n;i++)
    {
        if (color[i] == 1) cnt1++;
        else cnt2++;
    }

    printf("%I64d
",cnt1 * cnt2 - (n - 1));

    return 0;
}

PS：记得要用long long，要不会wa。