深度优先遍历和 宽度优先遍历

写在前面

DFS和BFS这是数据结构 最基本的两个算法,这里记录两道题的求解过程复习一下这两道算法

克隆图(leetcode133)

这题没什么说的,就是给你一个图,给出这个图的深度复制结果
在这里插入图片描述

深度优先

对于深度优先算法,最重要的是使用一个哈希表来维持visited过的点,具体实现见代码:

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector<Node*> neighbors;
    
    Node() {
        val = 0;
        neighbors = vector<Node*>();
    }
    
    Node(int _val) {
        val = _val;
        neighbors = vector<Node*>();
    }
    
    Node(int _val, vector<Node*> _neighbors) {
        val = _val;
        neighbors = _neighbors;
    }
};
*/

class Solution {
public:
    unordered_map<Node*, Node*>visited;
    Node* cloneGraph(Node* node) {
        if (!node)return node;
        Node* newnode = new Node(node->val, vector<Node*>());
        if (visited.find(node) != visited.end()) {
            return visited.find(node)->second;
        }
        visited[node] = newnode;
        for (int i = 0; i < node->neighbors.size(); i++) {
            if (visited.find(node->neighbors[i])==visited.end()) {
                //visited[node->neighbors[i]] = new Node(node->neighbors[i]->val, vector<Node*>());
                cloneGraph(node->neighbors[i]);
            }
            newnode->neighbors.push_back(visited.find(node->neighbors[i])->second);
        }
        return newnode;
    }

};

注意

  1. visited的更新,是在for循环之外,for循环只是逐个调用这个递归的函数,不是在for循环里面执行
  2. unordered_map的用法
    • find如果找不到,会等于 visited.end()
    • 如果找到了,返回的是一个iterator,调用键值可以直接->second这样子
    • 存储可以直接visited[a]=b这样子

广度优先

广度优先的精华在于另外维护一个队列,每次从队列中取一个元素来遍历,区别在于,这样要在for循环中更新 visited
上代码

class Solution {
public:
    
    Node* cloneGraph(Node* node) {
        if (!node) return node;
        unordered_map<Node*, Node*>visited;
        queue<Node*>node_queue;
        Node* newnode = new Node(node->val, vector<Node*>());
        visited[node] = newnode;
        node_queue.push(node);
        while(!node_queue.empty()) {
            Node *tmp = node_queue.front();
            node_queue.pop();
            for (int i = 0; i < tmp->neighbors.size(); i++) {
                if (visited.find(tmp->neighbors[i]) == visited.end()) {
                    node_queue.push(tmp->neighbors[i]);
                    visited[tmp->neighbors[i]] = new Node(tmp->neighbors[i]->val, vector<Node*>());
                }
                visited.find(tmp)->second->neighbors.push_back(visited.find(tmp->neighbors[i])->second);
            }
        }
        return newnode;
    }

};

注意

  1. queue的用法,front和pop要分开使用
  2. visited在for循环内部更新

岛屿数量(leetcode200)

题目描述:
在这里插入图片描述

深度优先

这题也是很容易想到要图遍历,不过这个是需要从每一个点开始出发,如果是标注为1的点,那么对应岛屿数量就要加一个,对每一个岛屿进行处理,上下左右遍历,能走的都要走遍
注意这题可以把grid直接当成visited来使用,只要是1那么就是没有访问过的,访问后置为0,就可以省下一部分空间

上代码

class Solution {
public:
    vector<vector<int>>directions = {{0,1}, {1,0}, {-1, 0}, {0, -1}};
    vector<vector<int>>visited;
    int row;
    int col;

    int numIslands(vector<vector<char>>& grid) {
        if (grid.size() <= 0)return 0;
        int row = grid.size();
        int col = grid[0].size();
        this->row = row;
        this->col = col;
        vector<vector<int>>visited(row, vector<int>(col));
        
        int count = 0;
        this->visited = visited;
        for (int i = 0; i < grid.size(); i++) {
            for (int j = 0; j < grid[0].size(); j++) {
                //visited[i][j] = 0;
                if (grid[i][j] == '1' && !this->visited[i][j]){
                    count++;
                    dfs(grid, i, j);
                }
            }
        }
        return count;
    }
    bool isIn(int x, int y) {
        return x>=0 && x <this->row && y>=0 &&y<this->col;
    }
    void dfs(vector<vector<char>>& grid, int oldi, int oldj) {
        this->visited[oldi][oldj] = 1;
        for (int i = 0; i < 4; i ++) {
            int newx = oldi + directions[i][0];
            int newy = oldj + directions[i][1];
            if (isIn(newx, newy) && grid[newx][newy] == '1' && !visited[newx][newy]){
                dfs(grid, newx,newy);
            }
        }
    }
};

广度优先

这个广度优先,和第133题其实差不多,就是又学会了一招pair和相应的second, first,废话不说,直接上代码:

class Solution {
public:
    vector<vector<int>>directions = {{0,1}, {1,0}, {-1, 0}, {0, -1}};
    vector<vector<int>>visited;
    int row;
    int col;

    int numIslands(vector<vector<char>>& grid) {
        if (grid.size() <= 0)return 0;
        int row = grid.size();
        int col = grid[0].size();
        


        int count = 0;
        
        
        for (int i = 0; i < grid.size(); i++) {
            for (int j = 0; j < grid[0].size(); j++) {
                if (grid[i][j] == '1' ){
                    count++;
                    queue<pair<int, int>>node_queue;
                    node_queue.push({i,j});
                    while(!node_queue.empty()) {
                        auto kk = node_queue.front();
                        node_queue.pop();
                        int r=kk.first;
                        int c = kk.second;
                        if (r + 1 < row && grid[r + 1][c] == '1') {
                            node_queue.push({r + 1, c});
                            grid[r + 1][c] = '0';
                        }
                        if (r - 1 >= 0 && grid[r - 1][c] == '1') {
                            node_queue.push({r - 1, c});
                            grid[r - 1][c] = '0';
                        }
                        if (c + 1 < col && grid[r][c + 1] == '1') {
                            node_queue.push({r, c + 1});
                            grid[r][c + 1] = '0';
                        }
                        if (c - 1 >= 0 && grid[r][c - 1] == '1') {
                            node_queue.push({r, c - 1});
                            grid[r][c - 1] = '0';
                        }
                    }
                }
            }
        }
        return count;
    }
    
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值