Leetcode 1466. 重新规划路线正向+逆向DFS/BFS

最新推荐文章于 2023-12-07 02:23:04 发布

自信的小螺丝钉

最新推荐文章于 2023-12-07 02:23:04 发布

阅读量183

点赞数

分类专栏： Leetcode 文章标签：深度优先 leetcode 宽度优先算法图论

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

版权

Leetcode 专栏收录该内容

324 篇文章 0 订阅

订阅专栏

原题链接：Leetcode 1466. 重新规划路线

在这里插入图片描述

BFS

class Solution {
public:
    vector<vector<int>> adj;
    vector<vector<int>> in;
    vector<int> visit;
    int minReorder(int n, vector<vector<int>>& connections) {
        adj.resize(n);
        visit.resize(n);
        in.resize(n);
        for(auto x:connections)
        {
            int a=x[0],b=x[1];
            adj[a].push_back(b);
            in[b].push_back(a);
        }
        int res=0;
        queue<int> q; 
        q.push(0);
        visit[0]=1;
        while(!q.empty())
        {
            int now=q.front(); q.pop();
            for(auto x:adj[now])
            {
                if(!visit[x])
                {
                    res++;
                    visit[x]=1;
                    q.push(x);
                }
            }
            for(auto x:in[now])
            {
                if(!visit[x])
                {
                    visit[x]=1;
                    q.push(x);
                }
            }
        }
        return res;
    }
};

DFS

class Solution {
public:
    vector<vector<int>> adj;
    vector<vector<int>> in;
    vector<int> visit;
    int res=0;
    void dfs(int now)
    {
        visit[now]=1;
        for(auto x:adj[now])
        {
            if(!visit[x])
            {
                res++;
                dfs(x);
            }
        } 
        for(auto x:in[now])
        {
            if(!visit[x]) dfs(x);
        }
    }
    int minReorder(int n, vector<vector<int>>& connections) {
        adj.resize(n);
        visit.resize(n);
        in.resize(n);
        for(auto x:connections)
        {
            int a=x[0],b=x[1];
            adj[a].push_back(b);
            in[b].push_back(a);
        }
        dfs(0);
        return res;
    }
};