1. 程式人生 > >資料結構實驗之圖論四:迷宮探索__DFS

資料結構實驗之圖論四:迷宮探索__DFS

Problem Description
有一個地下迷宮,它的通道都是直的,而通道所有交叉點(包括通道的端點)上都有一盞燈和一個開關;請問如何從某個起點開始在迷宮中點亮所有的燈並回到起點?

Input
連續T組資料輸入,每組資料第一行給出三個正整數,分別表示地下迷宮的結點數N(1 < N <= 1000)、邊數M(M <= 3000)和起始結點編號S,隨後M行對應M條邊,每行給出一對正整數,表示一條邊相關聯的兩個頂點的編號。

Output
若可以點亮所有結點的燈,則輸出從S開始並以S結束的序列,序列中相鄰的頂點一定有邊,否則只輸出部分點亮的燈的結點序列,最後輸出0,表示此迷宮不是連通圖。
訪問頂點時約定以編號小的結點優先的次序訪問,點亮所有可以點亮的燈後,以原路返回的方式回到起點。

Sample Input
1
6 8 1
1 2
2 3
3 4
4 5
5 6
6 4
3 6
1 5

Sample Output
1 2 3 4 5 6 5 4 3 2 1

AC程式碼:

#include<bits/stdc++.h>
using namespace std;
int mp[3001][3001],vis[3001],dis[3001];
int k,m,t,s;
void DFS(int i)
{
    dis[s++]=i;
    for(int j=1; j<=k; j++)
    {
        if(mp[i][j]==1&&vis[j]==0)
        {
            vis[j]=1;
            DFS(j);
            dis[s++]=i;
        }
    }
}
int main()
{
    int T,u,v;
    cin>>T;
    while(T--)
    {
        memset(mp,0,sizeof(mp));
        memset(vis,0,sizeof(vis));
        cin>>k>>m>>t;
        for(int i=0; i<m; i++)
        {
            cin>>u>>v;
            mp[u][v]=mp[v][u]=1;
        }
        s=0;
        vis[t]=1;
        DFS(t);
        for(int i=0; i<s; i++)
        {
            printf(i==s-1?"%d":"%d ",dis[i]);
        }
        if(s<2*k-1) cout<<" 0";
        cout<<endl;
    }
    return 0;
}

————
餘生還請多多指教!