ZOJ 3761 Easy billiards 月赛E DFS

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3761

题目大意：给定n个位置的小球，小球的运动方向只能是水平或者竖直的。按一定方向推动某球，当行径上有其他球时，停留在被撞球的位置，被撞的球沿原小球运动方向运动，当行径路径上没有其他球时，该小球被剔除。问：只能通过小球的相撞来剔除小球，最少能留下几个小球。

解题思路：首先，可以用并查集找出互相连通的小球集合个数。因为每一次推动的结果可以看成是被推动的小球被剔除了，其他位置上的小球不动，所以可以从每个集合中任选一个小球作DFS。选择推动小球的方向即为该球指向其父亲节点的方向。

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int set[2005];
int e,first[2005],next[4010],v[4010];
bool vis[2005];
struct node
{
    int x,y,l,id;
}p[2005],q[2005];
bool cmp1(node a,node b)
{
    if(a.x==b.x)
        return a.y<b.y;
    return a.x<b.x;
}
bool cmp2(node a,node b)
{
    if(a.y==b.y)
        return a.x<b.x;
    return a.y<b.y;
}
void link(int a,int b)
{
    v[e]=b;
    next[e]=first[a];
    first[a]=e++;
}
int find(int x)
{
    if(set[x]==x)
        return x;
    else return set[x]=find(set[x]);
}
void UN(int a,int b)
{
    int aa,bb;
    aa=find(a);
    bb=find(b);
    if(aa==bb)
        return;
    else
        set[aa]=set[bb];
}
void solve(int now)
{
    int k;
    for(int i=first[now];i!=-1;i=next[i])
    {
        k=v[i];
        if(!vis[k])
        {
            vis[k]=true;
            if(q[now].x==q[k].x)
            {
                if(q[now].y>q[k].y)//up
                    q[k].l=1;
                else
                    q[k].l=2;//down
            }
            else
            {
                if(q[now].x>q[k].x)//right
                    q[k].l=3;
                else
                    q[k].l=4;//left;
            }
            solve(k);
        }
    }
    if(q[now].l==4)
        printf("(%d, %d) LEFT
",q[now].x,q[now].y);
    else if(q[now].l==3)
        printf("(%d, %d) RIGHT
",q[now].x,q[now].y);
    else if(q[now].l==1)
        printf("(%d, %d) UP
",q[now].x,q[now].y);
    else if(q[now].l==2)
        printf("(%d, %d) DOWN
",q[now].x,q[now].y);
     return;    
}


int main()
{
    int n,j,i,lef[2001],ans;
    while(~scanf("%d",&n))
    {
        memset(first,-1,sizeof(first));
        memset(vis,false,sizeof(vis));
        e=0;ans=0;
        for(i=0;i<n;i++)
        {
            p[i].id=i;
            scanf("%d%d",&p[i].x,&p[i].y);
            q[i].x=p[i].x;q[i].y=p[i].y;
            q[i].l=-1;
        }
        for(i=0;i<n;i++)
            set[i]=i;
        sort(p,p+n,cmp1);//按x从小到大排序
        for(i=1;i<n;i++)
        {
            if(p[i].x==p[i-1].x)
            {
                link(p[i].id,p[i-1].id);
                link(p[i-1].id,p[i].id);
                UN(p[i].id,p[i-1].id);
            }
        }
        sort(p,p+n,cmp2);
        for(i=1;i<n;i++)
        {
            if(p[i].y==p[i-1].y)
            {
                link(p[i].id,p[i-1].id);
                link(p[i-1].id,p[i].id);
                UN(p[i].id,p[i-1].id);
            }
        }
        for(i=0;i<n;i++)
        {
            if(set[i]==i)
                lef[ans++]=i;
        }
        cout<<ans<<endl;
        for(i=0;i<ans;i++)
        {
            vis[lef[i]]=true;
            solve(lef[i]);
        }
        // cout<<e<<endl;
    }
    
    return 0;
}