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=0;i<=n;i++){
        f[i] = i;
        rank[i]=1;
    }
}

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