poj 1838

http://poj.org/problem?id=1838

并查集，，，计算总共个数的模版。。。。。

 #include <iostream>
 #define maxn 16006
 #include<algorithm>
 using namespace std;

 struct node{
     int x,y;
     int index;
 }q[maxn];

 int f[maxn],rank[maxn],sum[maxn];

 void init(int n){
     for(int i=;i<=n;i++){
         f[i] = i;
         rank[i]=;
     }
 }

 bool cmp_x(node a, node b){
     if(a.x==b.x)
         return a.y<b.y;
     return a.x<b.x;
 }

 bool cmp_y(node a,node b){
     if(a.y==b.y)
         return a.x<b.x;
     return a.y<b.y;
 }
 bool cmp(int x, int y){
     return x>y;
 }

 int find(int x){
     if(x!=f[x])
         f[x] = find(f[x]);
     return f[x];
 }

 void unin(int x, int y){
     int fx = find(x);
     int fy = find(y);
     if(fx==fy)
         return;
     if(rank[fx]>rank[fy]){
         rank[fx] += rank[fy];//求共有多少节点
         f[fy] = fx;
     }
     else {
         rank[fy] += rank[fx];
         f[fx] = fy;

     }
     return ;
 }

 int main()
 {
     int n , k;
     while(cin>>n>>k){
         init(n);
         for(int i=;i<=n;i++){
             cin>>q[i].x>>q[i].y;
             q[i].index=i;
         }
         sort(q+,q++n,cmp_x);//合并x方向
         for(int i=;i<n;i++){
             if(q[i].x==q[i+].x&&q[i].y+==q[i+].y)
                 unin(q[i].index,q[i+].index);
         }
         sort(q+,q++n,cmp_y);//合并y方向
         for(int i=;i<n;i++){
             if(q[i].y==q[i+].y&&q[i].x+==q[i+].x)
                 unin(q[i].index,q[i+].index);
         }
         int cnt =;
         for(int i=;i<=n;i++){//取出f[i]==i的节点
             if(f[i]==i)
                 sum[cnt++] = rank[i];
         }
         sort(sum,sum+cnt,cmp);//找出k个
         int ssum =;
         for(int i=;i<k;i++)
             ssum += sum[i];
         cout<<ssum<<endl;
     }
     return ;
 }