做题记录——图

Kephenny

已于 2023-03-02 15:17:32 修改

阅读量120

点赞数

文章标签： leetcode 算法数据结构

于 2022-07-23 21:12:06 首次发布

本文链接：https://blog.csdn.net/m0_72166786/article/details/125952767

版权

本文介绍了如何使用广度优先搜索（BFS）算法解决一道关于寻找给定图中颜色交替最短路径的问题。题目中，图的边分为红色和蓝色，目标是从节点0找到所有节点的红色和蓝色边交替的最短路径。通过建立邻接矩阵并进行BFS搜索，可以分别更新两种颜色路径的最短长度，并在搜索过程中确保路径交替。最终，对于每个节点，返回两种路径中较短的一种，若不存在交替路径，则返回-1。

摘要由CSDN通过智能技术生成

1.广度优先搜索

（1）lc1129：颜色交替的最短路径（2023/2/2）

题目：

给定一个整数 n，即有向图中的节点数，其中节点标记为 0 到 n - 1。图中的每条边为红色或者蓝色，并且可能存在自环或平行边。

给定两个数组 redEdges 和 blueEdges，其中：

redEdges[i] = [ai, bi] 表示图中存在一条从节点 ai 到节点 bi 的红色有向边，
blueEdges[j] = [uj, vj] 表示图中存在一条从节点 uj 到节点 vj 的蓝色有向边。

返回长度为 n 的数组 answer，其中 answer[X] 是从节点 0 到节点 X 的红色边和蓝色边交替出现的最短路径的长度。如果不存在这样的路径，那么 answer[x] = -1。

示例：

输入：n = 3, red_edges = [[0,1],[1,2]], blue_edges = []
输出：[0,1,-1]

思路：【官方题解】

作者：力扣官方题解
链接：https://leetcode.cn/problems/shortest-path-with-alternating-colors/solutions/2086543/yan-se-jiao-ti-de-zui-duan-lu-jing-by-le-vnuu/

使用 0 表示红色，1 表示蓝色，对于某一节点 x，从节点 0 到节点 x 的红色边和蓝色边交替出现的路径有两种类型：

类型 0：路径最终到节点 x 的有向边为红色；

类型 1：路径最终到节点 x 的有向边为蓝色。

如果存在从节点 0 到节点 x 的类型 0 的颜色交替路径，并且从节点 x 到节点 y 的有向边为蓝色，那么该路径加上该有向边组成了从节点 0 到节点 y 的类型 1 的颜色交替路径。类似地，如果存在从节点 0 到节点 x 的类型 1 的颜色交替路径，并且从节点 x 到节点 y 的有向边为红色，那么该路径加上该有向边组成了从节点 0 到节点 y 的类型 0 的颜色交替路径。

使用广度优先搜索获取从节点 0 到某一节点的两种类型的颜色交替最短路径的长度，广度优先搜索的队列元素由节点编号和节点路径类型组成，初始时节点 0 到节点 0 的两种类型的颜色交替最短路径的长度都是 0，将两个初始值入队。对于某一个队列元素，节点编号为 x，节点路径类型为 t，那么根据类型 t 选择颜色为 1−t 的相邻有向边，如果有向边的终点节点 y 对应类型 1−t 没有被访问过，那么更新节点 y 的类型 1−t 的颜色交替最短路径的长度为节点 x 的类型 t 的颜色交替最短路径的长度加 1，并且将它入队。

从节点 0 到某一节点的颜色交替最短路径的长度为两种类型的颜色交替最短路径长度的最小值。

代码：

class Solution {
public:
    vector<int> shortestAlternatingPaths(int n, vector<vector<int>>& redEdges, vector<vector<int>>& blueEdges) {
        vector<vector<vector<int>>> next(2, vector<vector<int>>(n));
        for (auto &e : redEdges) {
            next[0][e[0]].push_back(e[1]);
        }
        for (auto &e : blueEdges) {
            next[1][e[0]].push_back(e[1]);
        }

        vector<vector<int>> dist(2, vector<int>(n, INT_MAX)); // 两种类型的颜色最短路径的长度
        queue<pair<int, int>> q;
        dist[0][0] = 0;
        dist[1][0] = 0;
        q.push({0, 0});
        q.push({0, 1});
        while (!q.empty()) {
            auto [x, t] = q.front();
            q.pop();
            for (auto y : next[1 - t][x]) {
                if (dist[1 - t][y] != INT_MAX) {
                    continue;
                }
                dist[1 - t][y] = dist[t][x] + 1;
                q.push({y, 1 - t});
            }
        }
        vector<int> answer(n);
        for (int i = 0; i < n; i++) {
            answer[i] = min(dist[0][i], dist[1][i]);
            if (answer[i] == INT_MAX) {
                answer[i] = -1;
            }
        }
        return answer;
    }
};