Zoj 3761 Easy billiards dfs

在桌面上给你一些球， 你可以选择击任意一个球， 向4个方向， 但是要求是球运动过程中必须能装上另一个球， 也即至少在这个方向上还存在一个球。然后撞上一个球后，之前的一个球停在碰撞处， 被碰撞的求继续按撞他的球的方向运动直到离开桌面，然后问你桌面上最少能存在几个球， 输出击球顺序（坐标，方向）。

比赛的时候我的思路是吧能撞到的求连边然后从入度为0的点去撞其它球， 如果有环的话就随便撞一个就是了， 可是一wa到底。。一直没发现自己的问题， 于是赛后参考别人的代码写了dfs版本，另外， 我连边的时候对于每个点只连了离他最近的上下左右的点， 是一开始的写法需要用到的（另外似乎有并查集写法也需要这样处理建边）吗

#include <set>
#include <map>
#include <queue>
#include <stack>
#include <cmath>
#include <string>
#include <cctype>
#include <cstdio>
#include <vector>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <sstream>
#include <iostream>
#include <algorithm>
using namespace std;
#define ls id<<1,l,mid
#define rs id<<1|1,mid+1,r
#define OFF(x) memset(x,-1,sizeof x)
#define CLR(x) memset(x,0,sizeof x)
#define MEM(x) memset(x,0x3f,sizeof x)
typedef long long ll ;
typedef pair<int,int> pii ;
const int maxn = 2000 + 50 ;
const int inf = 0x3f3f3f3f ;
const int MOD = 1e9+7 ;

int x[maxn],y[maxn],n, vis[maxn];
int par[maxn],ran[maxn],indeg[maxn];
vector<int> G[maxn];

char s[4][10] = {"UP", "RIGHT", "DOWN", "LEFT"};

struct pnt {
    int x,y;
    int id;
//    pnt(int x,int y,int id):x(x),y(y),id(id){}
} p[maxn];
vector<pii> ans;

void add(int x, int y) {
    G[x].push_back(y);
    G[y].push_back(x);
    indeg[x]++;
    indeg[y]++;
}

bool cmp1(const pnt &a, const pnt &b) {
    if (a.x == b.x) return a.y < b.y;
    return a.x < b.x;
}

bool cmp2(const pnt &a, const pnt &b) {
    if (a.y == b.y) return a.x < b.x;
    return a.y < b.y;
}

bool cmp3(const pnt &a, const pnt &b) {
    return a.id < b.id;
}

void Dfs(int u) {
    vis[u] = 1;
    for (int i = 0; i < G[u].size(); i++) {
        int v = G[u][i];
        if (vis[v]) continue ;
        Dfs(v);
    }

}

int get_ans() {
    int res = 0;
    for (int i = 1; i <= n; i++) {
        if (vis[i]) continue ;
        res++;
        Dfs(i);
    }
    return res;
}

void build(){
    sort(p + 1, p + n + 1, cmp1);
    for (int i = 1; i < n; i++) {
        if (p[i].x == p[i + 1].x) {
//                cout << p[i].id << " " << p[i + 1].id << "\n" ;
            G[p[i].id].push_back(p[i + 1].id);
            G[p[i + 1].id].push_back(p[i].id);
        }
    }
    sort(p + 1, p + n + 1, cmp2);
    for (int i = 1; i < n; i++) {
        if (p[i].y == p[i + 1].y) {
            G[p[i].id].push_back(p[i + 1].id);
            G[p[i + 1].id].push_back(p[i].id);
        }
    }
    sort(p + 1, p + n + 1, cmp3);
    for (int i = 1; i <= n; i++) {
        indeg[i] = G[i].size();
    }
    CLR(vis);
    sort(p + 1, p + n + 1, cmp3);
}

void dfs(int u) {
    vis[u] = 1;
    for (int i = 0; i < G[u].size(); i++) {
        int v = G[u][i];
        if (vis[v]) continue;
        dfs(v);
        int dir;
        if (p[v].x == p[u].x) {
            if (p[v].y > p[u].y) dir = 2;
            else dir = 0;
        } else {
            if (p[v].x > p[u].x) dir = 3;
            else dir = 1;
        }
        ans.push_back(pii(v,dir));
    }
}

int main () {
#ifdef LOCAL
	freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
//      freopen("C:\\Users\\Administrator\\Desktop\\out.txt","w",stdout);
#endif
    while (~scanf("%d", &n)) {
        for (int i = 1; i <= n; i++) G[i].clear();
        CLR(indeg);
        for (int i = 1; i <= n; i++) {
            scanf("%d%d", &p[i].x, &p[i].y);
            p[i].id = i;
        }
        build();
        printf("%d\n",get_ans());
        CLR(vis);ans.clear();
        for (int i = 1; i <= n; i++) {
            if (vis[i]) continue ;
            dfs(i);
        }
        for (int i = 0; i < ans.size(); i++) printf("(%d, %d) %s\n",p[ans[i].first].x, p[ans[i].first].y, s[ans[i].second]);
    }
	return 0;
}