Leetcode 1129. 颜色交替的最短路径 BFS+模拟

最新推荐文章于 2024-06-12 10:00:23 发布

自信的小螺丝钉

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量120

点赞数

分类专栏： Leetcode 文章标签： leetcode 宽度优先算法数据结构图论

本文链接：https://blog.csdn.net/qq_45791939/article/details/128105074

版权

Leetcode 专栏收录该内容

324 篇文章 0 订阅

订阅专栏

原题链接：Leetcode 1129. 颜色交替的最短路径

在这里插入图片描述

参考：颜色交替的最短路径思路清晰基于基本BFS轻松写出充分理解题意

class Solution {
public:
    vector<int> shortestAlternatingPaths(int n, vector<vector<int>>& redEdges, vector<vector<int>>& blueEdges) {
        vector<vector<pair<int,int>>> g(n);
        vector<vector<int>> visit(n,vector<int>(2,0));
        for(auto x:redEdges)
        {
            int a=x[0],b=x[1];
            g[a].push_back({b,0});
        }
        for(auto x:blueEdges)
        {
            int a=x[0],b=x[1];
            g[a].push_back({b,1});
        }
        queue<vector<int>> q;
        vector<int> dis(n,INT_MAX);
        dis[0]=0;
        q.push({0,0,1});
        q.push({0,0,0});
        while(!q.empty())
        {
            auto tmp=q.front();
            int from=tmp[0];
            int d=tmp[1];
            int col=tmp[2];
            q.pop();
            for(auto x:g[from])
            {
                int to=x.first,color=x.second;
                if(col==(!color) && !visit[to][color]) 
                {
                    dis[to]=min(d+1,dis[to]);
                    q.push({to,d+1,color});
                    visit[to][color]=1;
                }
            }
        }
        for(int i=0;i<n;i++)
        {
            if(dis[i]==INT_MAX) dis[i]=-1;
        }
        return dis;
    }
};