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[];
 bool vis[];

 int color[];

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

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

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

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

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

 int main()
 {
     int n;

     scanf("%d",&n);

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

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

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

     dfs();

     long long cnt1 = ,cnt2 = ;

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

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

     return ;
 }

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