leetcode——815. 公交路线

思路

1. 如何建图？——每辆公交为点，到每个车站的距离
2. 如何扩展？——初始车站，有多少辆公交，公交所达的车站即可达
3. 如何获取答案？——到达目标车站的公交，与初始车站的距离 的最小值

代码

步骤：
1. 特殊情况处理——起始与终点相等
2. 优化建图  以每辆公交到车站是否可达  同时统计车站公交线路
3. 将原始车站拥有的公交压入队列 进行扩展
4. 对目标车站的公交进行遍历 查找最小值


class Solution {
public:
    int numBusesToDestination(vector<vector<int>>& routes, int source, int target) {
        //特殊情况处理
        if (source == target)
            return 0;
        //优化建图
        int n = routes.size();
        vector<vector<int>> edge(n, vector<int>(n));//距离矩阵
        unordered_map<int, vector<int>> rec;
        for (int i = 0; i < n; i++) {//遍历行
            for (int site : routes[i]) {//遍历列
                for (int j : rec[site]) {//遍历该车站可达的其他公交
                    edge[i][j] = edge[j][i] = true;//置公交车 到车站可达
                }
                rec[site].push_back(i);//该车站拥有的公交
            }
        }

        vector<int> dis(n, -1);
        queue<int> que;
        //将原始车站拥有的公交压入队列 并更新距离矩阵
        for (int bus : rec[source]) {
            dis[bus] = 1;
            que.push(bus);
        }
        //进行广度搜索 更新距离矩阵
        while (!que.empty()) {
            int x = que.front();que.pop();
            //如果该公交可到某个车站  且dis未被更新
            for (int y = 0; y < n; y++) {
                if (edge[x][y] && dis[y] == -1) {
                    dis[y] = dis[x] + 1;
                    que.push(y);//压入队列
                }
            }
        }
        
        int ret = INT_MAX;
        for (int bus : rec[target])//遍历到达目标车站的公交
            if (dis[bus] != -1)//如果该公交到起始车站的距离不为0  取最小值
                ret = min(ret, dis[bus]);
            
        return ret == INT_MAX ? -1 : ret;
    }
};