Leetcode 815.公交路线

公交路线

1.题目描述

我们有一系列公交路线。每一条路线 routes[i] 上都有一辆公交车在上面循环行驶。例如，有一条路线 routes[0] = [1, 5, 7]，表示第一辆 (下标为0) 公交车会一直按照 1->5->7->1->5->7->1->… 的车站路线行驶。

假设我们从 S 车站开始（初始时不在公交车上），要去往 T 站。 期间仅可乘坐公交车，求出最少乘坐的公交车数量。返回 -1 表示不可能到达终点车站。

示例：

输入：
routes = [[1, 2, 7], [3, 6, 7]]
S = 1
T = 6
输出：2
解释：
最优策略是先乘坐第一辆公交车到达车站 7, 然后换乘第二辆公交车到车站 6。

提示：

1 <= routes.length <= 500.
1 <= routes[i].length <= 10^5.
0 <= routes[i][j] < 10 ^ 6.

2.解体思路

解题思想为图广度优先搜索(BFS),问题难点在于如何建图，如何搜索？
图的表示为G=(N,E),即结点和边组成，此题由于0 <= routes[i][j] < 10 ^ 6所以选取车站为结点不合适，那么我们可以选取公交车为结点。
算法思想：

（1）构建图结点：以公交车为结点，其中车站为属性，如下图：

（2）构建边关系：结点i与结点j之间是否能互通，可以简化为，对于i节点中存在一车站，结点j中是否存在，若存在，则互通，否者，不通。

由于0号结点与1号结点共同拥有属性7号车站，所以0，1之间互通，则用例构建邻接矩阵为：

（3）构造访问结点集合与目标结点集合

访问结点结合：记录结点是否访问，针对图中存在环的问题。
目标结点集合：目标车站所在的公交车结点集合，目标结点结合中包括当前结点，则搜索完成。

（4）BFS搜索

主要使用队列来记录可以访问的结点，找最小换乘次数，就是S->T寻找最短路径，队列中保存两个元素，结点编号以及步数，由于步数从1开始，只要是搜索到目标结点，则一定是最短路径，因为，广度优先搜索，先遍历步数1的，然后步数2，····，所以一定先找到的是步数小的。

3.代码复现

#include<algorithm>
class Solution {
public:
    int numBusesToDestination(vector<vector<int>>& routes, int S, int T) {
        if(S==T) return 0;
        int N=routes.size();
        for(int i=0;i<N;++i)
        {
            sort(routes[i].begin(),routes[i].end());  //由于后面使用二分查找，所以先排序
        }
        vector<vector<int>> graph(N);     //采用压缩存储方式
        for(int i=0;i<N;++i)
        {
            for(int j=i+1;j<N;++j)
            {
                if(connection(routes[i],routes[j]))
                {
                    graph[i].push_back(j);    //填充邻接矩阵
                    graph[j].push_back(i);
                }
            }
        }

        set<int> vis;   //访问结点集合
        set<int> target;    //目标结点集合
        queue<pair<int,int>> q;
        for(int i=0;i<N;++i)
        {
            if(binary_search(routes[i].begin(),routes[i].end(),S))
            {
                vis.insert(i);      //包括S车站的都为访问结点和起始点
                q.push(pair<int,int>{i,1});     //由于初始不在车上，所以初值设为1
            }
            if(binary_search(routes[i].begin(),routes[i].end(),T))
                target.insert(i);       //包括T车站的都为目标点
        }

        while(!q.empty())
        {
            int cur=q.front().first;    //当前结点编号
            int dis=q.front().second;     //当前步数
            q.pop();
            if(target.count(cur))  return dis;    //若目标点集合包括当前结点，则返回，一定最小
            for(int i=0;i<graph[cur].size();++i)
            {
                if(!vis.count(graph[cur][i]))    //与当前结点连通且为未访问过结点入队
                  {
                        q.push(pair<int,int>{graph[cur][i],dis+1});   //距离++
                        vis.insert(graph[cur][i]);             //增加到访问结点结合
                  }
            }
        }
        return -1;
    }

    bool connection(vector<int> &a,vector<int>&b)    //判断两个结点是否互通
    {
        int i=0,j=0;
        while(i<a.size()&&j<b.size())       //由于数组已完成排序，采用双指针搜索，提高效率
        {
            if(a[i]==b[j])
                return true;
            else if(a[i]>b[j])
                j++;
            else 
                i++;
        }
        return false;
    }
};

4.总结

此题考察点：问题的抽象能力以及应变能力，要想到图的结点为公交车，而不是车站。
BFS为基本算法，难度不大。