1392. Dreaming of Stars

http://acm.timus.ru/problem.aspx?space=1&num=1392

一个森林   每棵树上的节点为一组  然后按照每种顺序输出

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<stack>
#include<set>
#include<map>
#include<queue>
#include<algorithm>
#include<cmath>
#define LL long long
//#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;

const int N=1005;
const int INF=0x3f3f3f3f;
//typedef pair<int,int>point;
int head[N],I;
struct node
{
    int j,next;
}side[N*N*2];
struct point
{
    int x,y,z;
    int r;
}mem[N];
vector<int>vt[N];
bool visited[N];
void add(int i,int j)
{
    side[I].j=j;
    side[I].next=head[i];
    head[i]=I++;
}

bool Reachable(int i,int j)
{
    return sqrt(1.0*(mem[i].x-mem[j].x)*(mem[i].x-mem[j].x)+
    (mem[i].y-mem[j].y)*(mem[i].y-mem[j].y)+
    (mem[i].z-mem[j].z)*(mem[i].z-mem[j].z))<double(mem[i].r+mem[j].r);

    /*
    int d=0,sum=0;
    if(mem[i].x!=mem[j].x)
    {++sum;d=abs(mem[i].x-mem[j].x);}
    if(mem[i].y!=mem[j].y)
    {++sum;d=abs(mem[i].y-mem[j].y);}
    if(mem[i].z!=mem[j].z)
    {++sum;d=abs(mem[i].z-mem[j].z);}
    if(sum>1)
    return false;
    if(d>=mem[i].r+mem[j].r)
    return false;
    return true;
    */
}

void Init(int n)
{
    for(int i=0;i<n;++i)
    {
        scanf("%d %d %d %d",&mem[i].x,&mem[i].y,&mem[i].z,&mem[i].r);
        for(int j=0;j<i;++j)
        {
            if(Reachable(i,j))
            {add(i,j);add(j,i);}
        }
    }
}
void dfs(int x,int k)
{
   visited[x]=true;
   vt[k].push_back(x);
   for(int t=head[x];t!=-1;t=side[t].next)
   {
       int j=side[t].j;
       if(!visited[j])
       {
           dfs(j,k);
       }
   }
}
int main()
{
    //freopen("data.in","r",stdin);
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(head,-1,sizeof(head));
        I=0;
        Init(n);
        memset(visited,false,sizeof(visited));
        for(int i=0;i<n;++i)
        vt[i].clear();
        int m=0;
        for(int i=0;i<n;++i)
        if(!visited[i])
        dfs(i,m++);
        for(int i=0;i<m;++i)
        {
            sort(vt[i].begin(),vt[i].end());
            for(unsigned int j=0;j<vt[i].size();++j)
            {
                if(j>0)
                cout<<", ";
                cout<<vt[i][j];
            }
            cout<<endl;
        }

    }
}