leetcode 542. 01 Matrix 、663. Walls and Gates(lintcode) 、773. Sliding Puzzle 、803. Shortest Distance from All Buildings

542. 01 Matrix

https://www.cnblogs.com/grandyang/p/6602288.html

将所有的1置为INT_MAX,然后用所有的0去更新原本位置为1的值。

最短距离肯定使用bfs。

每次更新了值的地方还要再加入队列中 。

class Solution {
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        int m = matrix.size(),n = matrix[0].size();
        queue<pair<int,int>> q;
        for(int i = 0;i < m;i++){
            for(int j = 0;j < n;j++){
                if(matrix[i][j] == 0)
                    q.push(make_pair(i,j));
                else
                    matrix[i][j] = INT_MAX;
            }
        }
        while(!q.empty()){
            auto coordinate = q.front();
            q.pop();
            int x = coordinate.first,y = coordinate.second;
            for(auto dir : dirs){
                int new_x = x + dir[0];
                int new_y = y + dir[1];
                if(new_x < 0 || new_x >= matrix.size() || new_y < 0 || new_y >= matrix[0].size() || matrix[new_x][new_y] <= matrix[x][y] + 1)
                    continue;
                matrix[new_x][new_y] = matrix[x][y] + 1;
                q.push(make_pair(new_x,new_y));
            }
        }
        return matrix;
    }
private:
    vector<vector<int>> dirs{{-1,0},{1,0},{0,-1},{0,1}};
};

 

663. Walls and Gates

https://www.cnblogs.com/grandyang/p/5285868.html

这个题跟542. 01 Matrix很像,主要利用bfs。先把所有的0位置放入队列中,然后通过0更新周围的位置达到更新所有的位置。

class Solution {
public:
    /**
     * @param rooms: m x n 2D grid
     * @return: nothing
     */
    void wallsAndGates(vector<vector<int>> &rooms) {
        // write your code here
        queue<pair<int,int>> q;
        for(int i = 0;i < rooms.size();i++){
            for(int j = 0;j < rooms[0].size();j++){
                if(rooms[i][j] == 0)
                    q.push(make_pair(i,j));
            }
        }
        while(!q.empty()){
            int x = q.front().first;
            int y = q.front().second;
            q.pop();
            for(auto dir : dirs){
                int x_new = x + dir[0];
                int y_new = y + dir[1];
                if(x_new < 0 || x_new >= rooms.size() || y_new < 0 || y_new >= rooms[0].size() || rooms[x_new][y_new] < rooms[x][y] + 1)
                    continue;
                rooms[x_new][y_new] = rooms[x][y] + 1;
                q.push(make_pair(x_new,y_new));
            }
        }
        return;
    }
private:
    vector<vector<int>> dirs{{-1,0},{1,0},{0,-1},{0,1}};
};

 

773. Sliding Puzzle

https://www.cnblogs.com/grandyang/p/8955735.html

queue里面存储的是每次变换后的board的样子和当前新的board中0所在的坐标。

将队列一次清空完了之后才能更新res,因为这就是一次变换。

用set相当于决定什么时候停止循环

class Solution {
public:
    int slidingPuzzle(vector<vector<int>>& board) {
        int res = 0,m = board.size(),n = board[0].size();
        set<vector<vector<int>>> visited;
        vector<vector<int>> correct{{1,2,3},{4,5,0}};
        queue<pair<vector<vector<int>>,vector<int>>> q;
        for(int i = 0;i < m;i++){
            for(int j = 0;j < n;j++){
                if(board[i][j] == 0){
                    vector<int> tmp;
                    tmp.push_back(i);
                    tmp.push_back(j);
                    q.push(make_pair(board,tmp));
                }
            }
        }
        while(!q.empty()){
            for(int i = q.size();i > 0;i--){
                int x = q.front().second[0];
                int y = q.front().second[1];
                vector<vector<int>> board_new = q.front().first;
                if(board_new == correct)
                    return res;
                visited.insert(board_new);
                q.pop();
                for(auto dir : dirs){
                    int new_x = x + dir[0];
                    int new_y = y + dir[1];
                    if(new_x < 0 || new_x >= m || new_y < 0 || new_y >= n)
                        continue;
                    vector<vector<int>> cand = board_new;
                    swap(cand[x][y],cand[new_x][new_y]);
                    if(visited.find(cand) != visited.end())
                        continue;
                    vector<int> tmp;
                    tmp.push_back(new_x);
                    tmp.push_back(new_y);
                    q.push(make_pair(cand,tmp));
                }
            }
            res++;
        }
        return -1;
    }
private:
    vector<vector<int>> dirs{{-1,0},{1,0},{0,-1},{0,1}};
};

注意犯的错误:

 for(int i = q.size();i > 0;i--)不能写成for(int i = 0;i < q.size();i++),因为q的size在循环一次后,大小是缩小的,这样不能保证正确的次数。

 

 

 

803. Shortest Distance from All Buildings

https://www.cnblogs.com/grandyang/p/5297683.html

到所有建筑物的距离和最小,其实也就等于到每个建筑物最小距离的和。

以每个建筑物做bfs求最小的距离,然后用一个dis的vector保存,每当增加一个建筑物,在相应位置叠加当前建筑物的最小距离即可。

有可能有些点无法访问到某一些建筑物,所以通过一个count矩阵存储每个位置能访问到的建筑物的个数,最后再判断能否达到。

class Solution {
public:
    /**
     * @param grid: the 2D grid
     * @return: the shortest distance
     */
    int shortestDistance(vector<vector<int>> &grid) {
        // write your code here
        int m = grid.size(),n = grid[0].size();
        int res = INT_MAX,building = 0;
        vector<vector<int>> dist(m,vector<int>(n,0));
        vector<vector<int>> count = dist;
        for(int i = 0;i < m;i++){
            for(int j = 0;j < n;j++){
                if(grid[i][j] == 1){
                    building++;
                    queue<pair<int,int>> q;
                    vector<vector<bool>> visited(m,vector<bool>(n,false));
                    q.push(make_pair(i,j));
                    int pace = 0;
                    while(!q.empty()){
                        pace++;
                        for(int i = q.size();i > 0;i--){
                            int x = q.front().first;
                            int y = q.front().second;
                            q.pop();
                            for(auto dir : dirs){
                                int new_x = x + dir[0];
                                int new_y = y + dir[1];
                                if(new_x < 0 || new_x >= m || new_y < 0 || new_y >= n || visited[new_x][new_y]== true || grid[new_x][new_y] != 0)
                                    continue;
                                dist[new_x][new_y] += pace;
                                count[new_x][new_y] += 1;
                                visited[new_x][new_y] = true;
                                q.push(make_pair(new_x,new_y));
                            }
                        }
                    }
                }
            }
        }
        for(int i = 0;i < m;i++){
            for(int j = 0;j < n;j++){
                if(grid[i][j] == 0 && count[i][j] == building)
                    res = min(res,dist[i][j]);
            }
        }
        return res == INT_MAX ? -1 : res;
    }
private:
    vector<vector<int>> dirs{{-1,0},{1,0},{0,-1},{0,1}};
};

 

另一种写法:

class Solution {
public:
    /**
     * @param grid: the 2D grid
     * @return: the shortest distance
     */
    int shortestDistance(vector<vector<int> > &grid) {
        // write your code here
        int m = grid.size(),n = grid[0].size();
        vector<vector<int> > path(m,vector<int>(n,0));
        vector<vector<int> > count(m,vector<int>(n,0));
        int building = 0;
        for(int i = 0;i < m;i++){
            for(int j = 0;j < n;j++){
                if(grid[i][j] == 1){
                    building++;
                    vector<vector<bool> > visited(m,vector<bool>(n,false));
                    search(grid,i,j,path,count,visited);
                }
            }
        }
        int res = INT_MAX;
        for(int i = 0;i < m;i++){
            for(int j = 0;j < n;j++){
                if(grid[i][j] == 0 && count[i][j] == building)
                    res = min(res,path[i][j]);
            }
        }
        return res == INT_MAX ? -1 : res;
    }
    void search(vector<vector<int> > grid,int x,int y,vector<vector<int> >& path,vector<vector<int> >& count,vector<vector<bool> > visited){
        queue<pair<int,int> > q;
        q.push(make_pair(x,y));
        int pace = 0;
        while(!q.empty()){
            pace++;
            for(int i = q.size();i > 0;i--){
                int x0 = q.front().first;
                int y0 = q.front().second;
                q.pop();
                //visited[x0][y0] = true;
                for(int j = 0;j < dirs.size();j++){
                    int new_x = x0 + dirs[j][0];
                    int new_y = y0 + dirs[j][1];
                    if(new_x < 0 || new_x >= path.size() || new_y < 0 || new_y >= path[0].size() || visited[new_x][new_y] || grid[new_x][new_y] != 0)
                        continue;
                    path[new_x][new_y] += pace;
                    count[new_x][new_y]++;
                    visited[new_x][new_y] = true;
                    q.push(make_pair(new_x,new_y));
                }
            }
        }
    }
private:
    vector<vector<int> > dirs{{-1,0},{1,0},{0,-1},{0,1}};
};


//visited[new_x][new_y] = true;只能放在新生成的坐标的位置

 

posted @ 2019-05-29 22:11  有梦就要去实现他  阅读(226)  评论(0编辑  收藏  举报