745 C. Hongcow Builds A Nation codeforces （并查集）

#include<bits/stdc++.h>
using namespace std;
bool visit[1005];
int pre[1005],num;
struct node
{
int flag;
int num1;
int num2;
}arr[1005];
int find(int x)
{
int y=x;
while(y!=pre[y])
{
y=pre[y];
}
int temp;
while(pre[x]!=y)
{
temp=pre[x];
pre[x]=y;
x=temp;
}
return y;
}
bool cmp(node x,node y)
{
if(x.num1>y.num1)return true;
return false;
}
int main()
{
int n,m,k,x,y;
cin>>n>>m>>k;
for(int i=1;i<=k;i++)
{
cin>>x;
visit[x]=1;
}
for(int i=1;i<=n;i++)
pre[i]=i;
memset(arr,0,sizeof(arr));
while(m--)
{
cin>>x>>y;
if(x>y) swap(x,y);
int xx=find(x),yy=find(y);
if(xx!=yy)
{
pre[yy]=xx;
arr[xx].num2+=arr[yy].num2+1;
}
else
{
arr[xx].num2++;
}
}
int num=0;
for(int i=1;i<=n;i++)
{
x=find(i);
arr[x].num1++;
if(arr[x].flag==0)
arr[x].flag=visit[i];
}
sort(arr+1,arr+1+n,cmp);
int i,sum=0,sum1=0;
for(i=1;i<=n;i++)
{
if(arr[i].flag)
break;
}
for(int j=1;j<=i-1;j++)
{
arr[i].num1+=arr[j].num1;arr[i].num2+=arr[j].num2;
arr[j].num1=0;
}
for(int j=i+1;j<=n;j++)
{
if(arr[j].num1==0) break;
if(!arr[j].flag)
{
arr[i].num1+=arr[j].num1;arr[i].num2+=arr[j].num2;
arr[j].num1=0;
}
}

for(int j=1;j<=n;j++)
{
if(arr[j].num1==0)
continue;
sum+=arr[j].num1*(arr[j].num1-1)/2-arr[j].num2;
}
cout<<sum<<endl;
return 0;
}