CodeForces - 862B Mahmoud and Ehab and the bipartiteness 【DFS】

题意：

给一颗树，问添多少条边让这颗树变成完全二分图

思路：

因为完全二分图的边数就是两边(两个集合的顶点个数)相乘嘛

然后题目已经给出n-1条边了

所以ans = l * r - (n-1)

这里的 l 和 r 是二分图两个集合的顶点个数

那么最大的问题在于这个 l 和 r 怎么求

看了题解后发现就是用dfs去跑一遍记录左右个数就好了

因为题目给你的树是会遍历所有顶点的

举个样例2的例子

先建树后

a[1] = {2}

a[2] = {1,3}

a[3] = {2,4}

a[4] = {3,5}

a[5] = {4}

初始化vis[1] = 1 【l = 1,r = 0】 跑dfs

先跑a[1]里未访问过的元素 也就是2

vis[2] = 0 ---> vis[2] = 1 r++【l = 1, r = 1】

再跑a[2]里面未访问过的元素 3 ......

如果两者之间有边那就把这两个分到不同集合

然后这题不能直接开二维数组 要开vector<int>[maxn]

大概就这样了

#include <bits/stdc++.h>
#include <vector>
#define ll long long

using namespace std;

const int maxn = 100005;
vector<int> a[maxn];
int vis[maxn];
ll l,r;

void dfs(int x, int step) {
	for(int i = 0; i < a[x].size(); i++) {
		if(!vis[a[x][i]]) {
			vis[a[x][i]] = 1;
			if(step&1) l++;
			else r++;
			dfs(a[x][i], step+1);
		}
	}
}

int main() {
	int n;
	scanf("%d", &n);
	l = 1;
	r = 0;
	for(int i = 0; i < n-1; i++) {
		int x, y;
		scanf("%d%d", &x, &y);
		a[x].push_back(y);
		a[y].push_back(x);
	}
	vis[1] = 1;
	dfs(1, 0);
	ll ans = l * r - (n-1);
	printf("%lld\n", ans);
}