1131 Subway Map (30分)

PAT30分最后一题做完了，👴的青春结束了
一般最后一题都是dijkstra做，加上之前做的几题刚好是dij找相同最短路径+dfs遍历确认所以第一次做这题的时候绕进去了（那样是24分，一个点WA一个点TLE）。然后参考了一下柳婼大神的思路，一次dfs解决，比想象的要轻松很多
dfs过程中记录每个站点到起始点的最短距离。换乘次数的局部最优不能推导全局最优，但是距离是可以的。所以当当前到达该站点的距离大于已有的距离时返回。判断返回的时候不能把换乘次数考虑进去，因为距离才是第一标尺，换乘次数要到搜索到终点时再判断是否更新最终路径。

#include <bits/stdc++.h>
#define INF 9999999
using namespace std;
const int maxn=10010;
int m,n,s=0,e=0;
struct Node{
    int stationID=-1,lineID=-1;
};
vector<Node> g[maxn];
int d[maxn];
int minlen=INF,mintrans=INF;
vector<int> path,res;
void printres(){
    int beg=s,ed=-1,lastline=-1,len=res.size();
    for(int i=0;i<len-1;i++){
        int currline=-1;
        for(int j=0;j<g[res[i]].size();j++){
            if(g[res[i]][j].stationID==res[i+1]){
                currline=g[res[i]][j].lineID;
                break;
            }
        }
        if(i==0)lastline=currline;
        else if(lastline!=currline){
            printf("Take Line#%d from %04d to %04d.\n",lastline,beg,res[i]);
            lastline=currline;
            beg=res[i];
        }
    }
    printf("Take Line#%d from %04d to %04d.\n",lastline,beg,res.back());
}
void dfs(int curr,int len, int transtime,int line){
    path.push_back(curr);
    if(len>minlen||d[curr]<len)return;
    d[curr]=len;
    if(curr==e){
        if(len<minlen||(len==minlen&&transtime<mintrans)){
            //update
            mintrans=transtime;
            minlen=len;
            res=path;
        }
        return;
    }
    for(Node station:g[curr]){
        if(curr==s)dfs(station.stationID,len+1,transtime,station.lineID);
        else dfs(station.stationID,len+1,transtime+(station.lineID==line?0:1),station.lineID);
        path.pop_back();
    }
}

int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>m;
        Node lastStation;
        lastStation.lineID=i;
        for(int j=0;j<m;j++){
            if(j==0)cin>>lastStation.stationID;
            else{
                Node nd;
                nd.lineID=i;
                cin>>nd.stationID;
                g[lastStation.stationID].push_back(nd);
                g[nd.stationID].push_back(lastStation);
                lastStation=nd;
            }
        }
    }
    cin>>m;
    for(int i=0;i<m;i++){
        fill(d,d+maxn,INF);
        minlen=mintrans=INF;
        path.clear();
        res.clear();
        cin>>s>>e;
        dfs(s,0,0,-1);
        printf("%d\n",minlen);
        printres();
    }
}