dfs遍历每一个节点

时间:2022-04-10 16:08:44

搜索真的是一个很好用的算法,但是一直都是很弱,代码写的也是很弱。


在一个吝啬的国度里有N个城市,这N个城市间只有N-1条路把这个N个城市连接起来。现在,Tom在第S号城市,他有张该国地图,他想知道如果自己要去参观第T号城市,必须经过的前一个城市是几号城市(假设你不走重复的路)。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>

using namespace std;

const int maxn = 100010;
vector<int> G[maxn];
int pre[maxn];

void dfs(int x)
{
    int i;
    for(i=0;i<G[x].size();i++)
    {
        if(pre[G[x][i]])
            continue;
        pre[G[x][i]] = x;
        dfs(G[x][i]);
    }
}
int main(void)
{
    int n,s,i,T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&s);
        memset(pre,0,sizeof(pre));
        for(i=0;i<=maxn;i++)
            G[i].clear();
        pre[s] = -1;
        for(i=0;i<n-1;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            G[a].push_back(b);
            G[b].push_back(a);
        }
        dfs(s);
        for(i=1;i<=n;i++)
            printf("%d ",pre[i]);
        printf("\n");
    }

    return 0;
}
/*
1
10 1
1 9
1 8
8 10
10 3
8 6
1 2
10 4
9 5
3 7
*/