codeforces 744A

链接: http://codeforces.com/problemset/problem/744/A

题意:
现在有 n 个点 ，m 条边组成了一个无向图 ， 其中有 k 个特殊点， 这些点之间不能相连 ，问我可以再多加几条边?
分析：
设 连通子块的点有n个 ， 那么将会有 n(n-1)/2 条边 ， 所以尽量将点都加入到一个点最多的连通子块中（注意特殊点不能相连）。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
using namespace std;
const int N = 10000+10;
struct node
{
    int n,number,flag;
};
node fa[N];
bool vis[N];
int find_fa(int x);
void union_xy(int x, int y);
int main()
{
    int i,j,n,m,k,a,b,id;
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        memset(vis,false,sizeof(vis));
        for(i=1; i<=k; i++)
        {
           scanf("%d",&a);
           id = a;
           vis[a] = true;
        }

        for(i=1; i<=n; i++)
            fa[i].n = i , fa[i].number = 1,fa[i].flag=0;
        int nn = m;
        while(nn--)
        {
            scanf("%d%d",&a,&b);
            union_xy(a,b);
        }

        for(i=1; i<=n; i++)
        if(fa[i].number >  fa[id].number && vis[i])
            id = i; //找点最多的连通子块
        int ans = 0 , flag = 0;
        for(i=1; i<=n; i++)
           if(fa[i].n==i && vis[i]==false)
           {
              union_xy(id,i) ; //将点加入到子块中
           }
        for(i=1; i<=n; i++)
          if(vis[i] || fa[i].n==i)
             ans += fa[i].number*(fa[i].number-1)/2;
        printf("%d\n",ans-m); //所有的边减去已经有的边
    }
    return 0;
}
int find_fa(int x)
{
    if(fa[x].n == x)
        return x;
    return fa[x].n = find_fa(fa[x].n);
}
void union_xy(int x, int y)
{
    int fx = find_fa(x);
    int fy = find_fa(y);
    if(fx == fy) //如果父亲相同 ，子块中就没有新加入点
    {
          fa[fx].number;
          return ;
    }

    if(vis[fx]) //一个点加入到一个子块，那么该点的所有点都要加入加来
       fa[fx].number += fa[fy].number,  fa[fy].n = fx;
    else
       fa[fy].number += fa[fx].number,  fa[fx].n = fy;
}