Mahmoud and Ehab and the bipartiteness CodeForces - 862B 并查集

传送门

题目大意：给定一个整数n（1 <= n <= 10^5) , 然后给n-1行， 表示u和v有一条边链接， 所有的数分成两部分，在同一侧的数字不能链接，问除了题目给定的链接外还有多少链接。

解题思路：主要是求解两侧分别有几个数字，然后结果就是 a × b - n +1; a和b分别为两边的数量。分类用并查集很简单，可以参考经典并查集题目：食物连。分成两类后遍历数一下看看几个即可。

AC代码：

#include <iostream>
#include <cstdio>

using namespace std;

const int max_n = 100050;

int par[max_n+max_n];
int Rank[max_n+max_n];

void init_par()
{
for(int i=0; i<max_n*2; i++)
{
par[i] = i;
Rank[i] = 0;
}
}

int find_par(int n)
{
if(par
== n) return n;
return par
= find_par(par
);
}

void unit_par(int a, int b)
{
a = find_par(a);
b = find_par(b);

if(a == b) return;
if(Rank[a] < Rank[b]) par[a] = b;
else
{
par[b] = a;
if(Rank[a] == Rank[b]) Rank[a]++;
}
}

bool same(int a, int b)
{
return find_par(a) == find_par(b);
}

int main()
{
int n, u, v;
scanf("%d", &n);
if(n == 1)
{
printf("0\n");
return 0;
}
init_par();
for(int i=1; i<n; i++)
{
scanf("%d%d", &u, &v);
unit_par(u, v+max_n);
unit_par(v, u+max_n);
}
int a = 0, b = 0;
for(int i=1; i<=n; i++)
{
if(same(u, i)) a++;
else if(same(u, i+max_n)) b++;
}
printf("%lld\n", (long long) a*b - n+1);
return 0;
}