poj 1838 Banana（并查集）

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

题意：有一个直角坐标系，相邻的点可以相连成一块，而且可以将k块相连，求最多有几个点可以连成一块。

思路：先并查集，再排序即可

 

#include<stdio.h>

#include<string.h>

#include<math.h>

#include<iostream>

#include<algorithm>

#include<queue>

#include<stack>

#define mem(a,b) memset(a,b,sizeof(a))

#define ll __int64

#define MAXN 16000

#define INF 0x7ffffff

#define lson l,m,rt<<1

#define rson m+1,r,rt<<1|1

using namespace std;

struct Point

{

    int x,y;

    int id;

};

Point p[16000+10];

int  fa[16000+10];

int num[16000+10];

int cnt[16000+10];

int n;

int cmpx(Point a,Point b)

{

    if(a.x!=b.x) return a.x<b.x;

    return a.y<b.y;

}

int cmpy(Point a,Point b)

{

    if(a.y!=b.y) return a.y<b.y;

    return a.x<b.x;

}

int find(int a)

{

    if(fa[a]!=a) fa[a]=find(fa[a]);

    return fa[a];

}



void bincha(int a,int b)

{

    int x=find(a);

    int y=find(b);

    if(x==y) return ;

    fa[x]=fa[y];

    cnt[y]+=cnt[x];

    cnt[x]=0;

}

int main()

{

    int m;

    int i,j;

    while(cin>>n>>m)

    {



        for(i=1;i<=n;i++)

        {

            cnt[i]=1;

            fa[i]=p[i].id=i;

            scanf("%d%d",&p[i].x,&p[i].y);

        }

        sort(p+1,p+n+1,cmpx);

        for(i=1;i<n;i++)

        {

            if(p[i].x==p[i+1].x&&p[i+1].y-p[i].y==1&&fa[p[i+1].id]!=fa[p[i].id])

            {

               int pre=fa[p[i].id];

               int pre1=fa[p[i+1].id];

               fa[p[i+1].id]=fa[p[i].id];

               cnt[pre]+=cnt[pre1];

               cnt[pre1]=0;

            }

        }

        sort(p+1,p+n+1,cmpy);

        for(i=1;i<n;i++)

        {

            if(p[i].y==p[i+1].y&&p[i+1].x-p[i].x==1&&fa[p[i+1].id]!=fa[p[i].id])

            {

               bincha(p[i+1].id,p[i].id);

            }



        }

        j=0;

        for(i=1;i<=n;i++)

        {

            if(cnt[i]!=0)

            {

                num[j++]=cnt[i];

            }

        }

        sort(num,num+j);        

        j--;

        int ans=0;

        while(m--)

        {

            ans+=num[j--];

        }

        cout<<ans<<endl;

    }

    return 0;

}