BZOJ 3060: [Poi2012]Tour de Byteotia 并查集

前 $k$ 个节点形成的结构必定是森林，而 $[k+1,r]$ 之间肯定是都连上，而剩下的一个在 $[1,k],$一个在 $[k+1,r]$ 的节点就能连多少连多少即可.

Code: 

#include <bits/stdc++.h>  
#define N 1000005 
#define setIO(s) freopen(s".in","r",stdin) 
using namespace std;   
struct Edge
{
	int u,v; 
}e[N<<1]; 
int p[N]; 
int find(int x) 
{
	return p[x]==x?x:p[x]=find(p[x]); 
}
int main() 
{ 
	// setIO("input"); 
	int n,m,k,i,j,cnt=0; 
	scanf("%d%d%d",&n,&m,&k);   
	for(i=1;i<=n;++i) p[i]=i; 
	for(i=1;i<=m;++i) 
	{
		scanf("%d%d",&e[i].u,&e[i].v); 
		if(e[i].u>e[i].v) swap(e[i].u,e[i].v);       
		if(e[i].v<=k) 
		{
			int xx=find(e[i].u),yy=find(e[i].v); 
			if(xx!=yy) 
			{
				p[xx]=yy; 
				++cnt; 
			}
		}   
		else if(e[i].u>k) 
		{
			++cnt;
			int xx=find(e[i].u),yy=find(e[i].v); 
			p[xx]=yy;   
		}     
	}
	for(i=1;i<=m;++i) 
	{
		if(e[i].u<=k&&e[i].v>k) 
		{
			int xx=find(e[i].u),yy=find(e[i].v); 
			if(xx!=yy) p[xx]=yy,++cnt; 
		}
	}
	printf("%d\n",m-cnt);   
	return 0; 
}