图的遍历 | 1131地铁图： dfs复杂模拟题

这题在搞清楚思路绕过坑后，还是可以写的出通过sample data的代码的。但是不能AC，让我很气。

最后查清原因：还是对dfs本质理解的不够。

wa代码：

vis[s]=1;
dfs(s,e,0);

殊不知本题有多个查询数据。如果只调用一遍还可以蒙混过关，但是这样的错误必然导致wa

ac代码：

vis[s]=1;
dfs(s,e,0);
vis[s]=0;

参考柳诺博客修改的AC代码：

#include <stdio.h>
#include <memory.h>
#include <math.h>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <queue>
#include <algorithm>
#include <map>
 
#define I scanf
#define OL puts
#define O printf
#define F(a,b,c) for(a=b;a<c;a++)
#define FF(a,b) for(a=0;a<b;a++)
#define FG(a,b) for(a=b-1;a>=0;a--)
#define LEN 10000
#define MAX 0x06FFFFFF
#define V vector<int>
 
using namespace std;
 
vector<int> g[LEN];
int line[LEN][LEN];
int vis[LEN];
vector<int> path;
vector<int> ans;
int min_d=MAX;
int min_ls=MAX;
 
int N,M,K;
 
int calc_ls(){
    int cnt=-,preLine=;
    for(int i=;i<path.size();i++){
        if(line[path[i-]][path[i]]!=preLine) cnt++;
        preLine=line[path[i-]][path[i]];
    }
    return cnt;
}
 
void dfs(int s,int e,int d){
    if(s==e){
        int ls=calc_ls();
        if(d<min_d || (d==min_d && ls<min_ls)){
            min_d=d;
            min_ls=ls;
            ans=path;
        }
        return;
    }
    int i;
    FF(i,g[s].size()){
        int o=g[s][i];
        if(!vis[o]){
            vis[o]=;
            path.push_back(o);
            dfs(o,e,d+);
            vis[o]=;
            path.pop_back();
        }
    }
 
}
 
void printLine(){
    int s=ans[];
    int preL=;
    int i;
    F(i,,ans.size()){
        if(line[ans[i-]][ans[i]]!=preL){
            if(preL) printf("Take Line#%d from %04d to %04d.\n",preL,s,ans[i-]);
            s=ans[i-];
            preL=line[ans[i-]][ans[i]];
        }
    }
    printf("Take Line#%d from %04d to %04d.\n",preL,s,ans[i-]);
}
 
int main(){
//    freopen("1131.txt","r",stdin);
    int s,e,i,j;
    I("%d",&N);
    F(i,,N+){
        int pre,p=-;
        I("%d",&M);
        while(M--){
            pre=p;
            I("%d",&p);
            if(pre>=){
                g[p].push_back(pre);
                g[pre].push_back(p);
                line[p][pre]=i;
                line[pre][p]=i;
            }
        }
    }
    I("%d",&K);
    while(K--){
        min_d=MAX;
        min_ls=MAX;
        path.clear();
        ans.clear();
        I("%d%d",&s,&e);
        path.push_back(s);
        vis[s]=;
        dfs(s,e,);
        vis[s]=;
        O("%d\n",min_d);
        printLine();
    }
    return ;
}

在自己思路上修改的AC代码：（个人认为比柳诺的好理解）

 #include <stdio.h>
 #include <memory.h>
 #include <math.h>
 #include <string>
 #include <vector>
 #include <set>
 #include <stack>
 #include <queue>
 #include <algorithm>
 #include <map>
 
 #define I scanf
 #define OL puts
 #define O printf
 #define F(a,b,c) for(a=b;a<c;a++)
 #define FF(a,b) for(a=0;a<b;a++)
 #define FG(a,b) for(a=b-1;a>=0;a--)
 #define LEN 10000
 #define MAX 0x06FFFFFF
 #define V vector<int>
 
 using namespace std;
 
 vector<int> g[LEN];
 int line[LEN][LEN];
 int vis[LEN];
 vector<int> path;
 vector<int> ans;
 int min_d=MAX;
 int min_ls=MAX;
 
 int N,M,K;
 
 void dfs(int s,int e,int d,int l,int ls){
     if(s==e){
         if(d<min_d || (d==min_d && ls<min_ls)){
             min_d=d;
             min_ls=ls;
             ans=path;
         }
         return;
     }
     int i;
     FF(i,g[s].size()){
         int o=g[s][i];
         if(!vis[o]){
             vis[o]=;
             path.push_back(o);
             int nl=line[s][o];
             int nls=ls;
             if(l==){    //初始结点
                 nls=;
             }else{
                 if(nl!=l) nls++;
             }
             dfs(o,e,d+,nl,nls);
             vis[o]=;
             path.pop_back();
         }
     }
 
 }
 
 void printLine(){
     int s=ans[];
     int preL=line[s][ans[]];
     int i;
     F(i,,ans.size()){
         if(line[ans[i-]][ans[i]]!=preL){
             printf("Take Line#%d from %04d to %04d.\n",preL,s,ans[i-]);
             s=ans[i-];
             preL=line[ans[i-]][ans[i]];
         }
     }
     printf("Take Line#%d from %04d to %04d.\n",preL,s,ans[i-]);
 }
 
 int main(){
 //    freopen("1131.txt","r",stdin);
     int s,e,i,j;
     I("%d",&N);
     F(i,,N+){
         int pre,p=-;
         I("%d",&M);
         while(M--){
             pre=p;
             I("%d",&p);
             if(pre>=){
                 g[p].push_back(pre);
                 g[pre].push_back(p);
                 line[p][pre]=i;
                 line[pre][p]=i;
             }
         }
     }
     I("%d",&K);
     while(K--){
         min_d=MAX;
         min_ls=MAX;
         path.clear();
         ans.clear();
         I("%d%d",&s,&e);
         vis[s]=;
         dfs(s,e,,,);
         vis[s]=;
         O("%d\n",ans.size());
         ans.insert(ans.begin(),s);
         printLine();
     }
     return ;
 }

注意点：

① 38 39 行，对维护的最小距离和最小换乘次数进行更新，不要写错（我开始写成了d=min_d ，查了很久的错，蠢哭……）

② 99 100 行，将最小距离和最小换乘次数重新初始化为INF。

③ 106 行，牢记 dfs 结构