codeforces A. Hongcow Builds A Nation 并查集

codeforces A. Hongcow Builds A Nation 

并查集 

题意 给定一张无向图
其中 有 k 个特殊点
特殊点之间不能存在 路径求 最多能加多少边
不能有重边自环
将各个连通块在没有加边的时候的点数 算出来
然后 将自由的点 都加入 到 点数最多的连通块中，
然后算出加边以后的总边数
减去原有的边数就是答案了

#include <bits/stdc++.h> 
using namespace std ; 

const int N = 1011 ; 
struct node{
    int n,number,flag ; 
}fa[N];
int n,m,k,sum,id ; 
bool vis[N] ; 


inline int find(int x) 
{
    if(fa[ x ].n ==x ) return x ; 
    return fa[ x ].n = find(fa[ x ].n) ; 
}

inline void Union(int x,int y) 
{
    int fx = find(x) ; 
    int fy = find(y) ; 
    if(fx==fy) return  ; 
    if(vis[fy]) swap(fx,fy) ; 
    
    if(vis[fx]) 
        fa[ fx ].number+=fa[ fy ].number ,fa[ fy ].n = fx ; 
    else 
        fa[ fy ].number+=fa[ fx ].number ,fa[ fx ].n = fy ;  
}

int main() 
{
    while(~scanf("%d%d%d",&n,&m,&k)) 
    {
        for(int i=1;i<N;i++) vis[ i ] = 0 ; 
        int x,y ; 
        for(int i=1;i<=n;i++) 
            fa[ i ].n = i ,fa[ i ].number = 1 ,fa[ i ].flag = 0 ; 
            
        for(int i=1;i<=k;i++) 
        {
            scanf("%d",&id) ; 
            vis[id] = 1 ; 
        }
        
        for(int i=1;i<=m;i++) 
        {
            scanf("%d%d",&x,&y) ; 
            Union(x,y) ; 
        }
        
        for(int i=1;i<=n;i++) 
            if( vis[ i ] && fa[ i ].number > fa[ id ].number ) id = i ; 
        for(int i=1;i<=n;i++) 
            if( !vis[ i ] && fa[ i ].n==i ) Union( id,i ) ; 
        sum = 0 ; 
        for(int i=1;i<=n;i++) 
            if(vis[ i ] || fa[ i ].n==i) 
                sum+=(fa[ i ].number-1) * fa[ i ].number /2 ; 
        printf("%d
",sum - m ) ;  
         
    }
    return 0 ; 
}