重返天梯-L3-014 周游世界 (30 分)(dfs)

题目描述

周游世界是件浪漫事，但规划旅行路线就不一定了…… 全世界有成千上万条航线、铁路线、大巴线，令人眼花缭乱。所以旅行社会选择部分运输公司组成联盟，每家公司提供一条线路，然后帮助客户规划由联盟内企业支持的旅行路线。本题就要求你帮旅行社实现一个自动规划路线的程序，使得对任何给定的起点和终点，可以找出最顺畅的路线。所谓“最顺畅”，首先是指中途经停站最少；如果经停站一样多，则取需要换乘线路次数最少的路线。
重返天梯-L3-014 周游世界 (30 分)

子问题拆解

维护双变量换乘输出

给定一个line[i][j]，表示从i到j站点所乘坐的线路
path[]表示所经过的所有站点（其中有换乘）

请输出每条换乘线路的首尾结点？

算复杂也不算复杂的一个问题，需要想到维护两个变量preLine,preTrans
往后遍历路径，如果线路变化发生换乘，再进行输出

// 维护两个变量，记录线路编号和这条线路的首站
int preLine = 0, preTrans = start;
for (int j = 1; j < path.size(); j++) {
    if (preLine != line[path[j-1]][path[j]]) {
        if (preLine != 0) printf("take the bus%d from %d to %d\n", preLine, preTrans, path[j-1]);
        preLine = line[path[j-1]][path[j]];
        preTrans = path[j-1];
    }
}

排列型枚举 dfs 最顺畅路径

void dfs(int node, int ed, int cnt, int &minCnt, int &minTrans) {
    if (node == ed && (cnt < minCnt || (cnt == minCnt && calTrans(tempPath) < minTrans))) {
        minCnt = cnt;
        minTrans = calTrans(tempPath);
        path = tempPath;
    }
    
    if (node == ed) return;
    
    for (int i = 0; i < v[node].size(); i++) {
        if (!st[v[node][i]]) {
            st[v[node][i]] = true;
            tempPath.push_back(v[node][i]);
            dfs(v[node][i], ed, cnt+1, minCnt, minTrans);
            st[v[node][i]] = false; // 还原现场
            tempPath.pop_back();
        }
    }
}

C++

#include <iostream>
#include <cstring>
#include <vector>
#define debug(x) cout << #x << '=' << x << endl
using namespace std;
const int N = 10010;
int n;
vector<vector<int>> v(N);
int line[N][N]; // 存储站点间的线路
bool st[N];
vector<int> tempPath, path;

int calTrans(vector<int> a) {
    int cnt = 0, preLine = 0;
    for (int j = 1; j < a.size(); j++) {
        if (line[a[j-1]][a[j]] != preLine) cnt++;
        preLine = line[a[j-1]][a[j]];
    }
    cnt--;
    return cnt;
}


void dfs(int node, int ed, int cnt, int &minCnt, int &minTrans) {
    if (node == ed) {
        if (cnt < minCnt || (cnt == minCnt && calTrans(tempPath) < minTrans)) {
            minCnt = cnt;
            minTrans = calTrans(tempPath);
            path = tempPath;
        }
        return;
    }
    
    for (int i = 0; i < v[node].size(); i++) {
        if (!st[v[node][i]]) {
            st[v[node][i]] = true;
            tempPath.push_back(v[node][i]);
            dfs(v[node][i], ed, cnt+1, minCnt, minTrans);
            st[v[node][i]] = false; // 还原现场
            tempPath.pop_back();
        }
    }
}


int main() {
    scanf("%d", &n);
    
    for (int i = 1; i <= n; i++) {
        int m, pre;
        scanf("%d%d", &m, &pre);
        for (int j = 1; j < m; j++) {
            int temp;
            scanf("%d", &temp);
            v[pre].push_back(temp); // 该站点可以到哪些站点
            v[temp].push_back(pre);
            line[pre][temp] = i, line[temp][pre] = i; 
            pre = temp;
        }
    }
    
    int k;
    scanf("%d", &k);
    
    while (k--) {
        int a, b;
        scanf("%d%d", &a, &b);
        int minCnt = 1e7, minTrans = 1e7;
        tempPath.clear();
        tempPath.push_back(a);
        st[a] = true;
        dfs(a, b, 0, minCnt, minTrans);
        st[a] = false;
        
        if (minCnt == 1e7) {
            puts("Sorry, no line is available.");
            continue;
        }
        
        cout << minCnt << endl;
        int preLine = 0, preTrans = a;
        for (int j = 1; j < path.size(); j++) {
            if (line[path[j-1]][path[j]] != preLine) {
                if (preLine != 0) 
                    printf("Go by the line of company #%d from %04d to %04d.\n", preLine, preTrans, path[j-1]);
                preLine = line[path[j-1]][path[j]];
                preTrans = path[j-1];
            }
        }
        printf("Go by the line of company #%d from %04d to %04d.\n", preLine, preTrans, b);
        
    }
    
    return 0;
}