吝啬的国度 简单的图搜索

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
int pre[100005];
typedef struct edge
{
    int v;
    struct edge *next;
}edge;
struct height{
    int num;
    struct edge *next;
}head[200010];

void add(int a,int b)//加边
{
    edge *ss=(edge *)malloc(sizeof(edge));
    ss->v=b;
    ss->next=head[a].next;
    head[a].next=ss;
}

void dfs(int cur)   //搜索
{
    edge *s;
    for(s=head[cur].next;s!=NULL;s=s->next)
    {
        if(pre[s->v])//标记搜索过了，并且找到当前的v的前驱；
            continue;
        pre[s->v]=cur;
        dfs(s->v);
    }
}
    
int main()
{
    int m,n,s,a,b;
    scanf("%d",&m);
    while(m--)
    {
        memset(head,NULL,sizeof(head));
        memset(pre,0,sizeof(pre));
        scanf("%d%d",&n,&s);
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            add(a,b);   add(b,a);
        }
        pre[s]=-1;
        dfs(s);
        for(int i=1;i<n;i++)
            printf("%d ",pre[i]);
        printf("%d
",pre[n]);
    }
    return 0;
}