1. 程式人生 > >佇列&棧//01 矩陣

佇列&棧//01 矩陣

給定一個由 0 和 1 組成的矩陣,找出每個元素到最近的 0 的距離。

兩個相鄰元素間的距離為 1 。

示例 1: 
輸入:

0 0 0
0 1 0
0 0 0

輸出:

0 0 0
0 1 0
0 0 0

示例 2: 
輸入:

0 0 0
0 1 0
1 1 1

輸出:

0 0 0
0 1 0
1 2 1

注意:

  1. 給定矩陣的元素個數不超過 10000。
  2. 給定矩陣中至少有一個元素是 0。
  3. 矩陣中的元素只在四個方向上相鄰: 上、下、左、右。
class Solution {
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        int m = matrix.size(), n = matrix[0].size();
        vector<vector<int>> dirs{{0,-1},{-1,0},{0,1},{1,0}};
        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({i,j});
                else matrix[i][j] = INT_MAX;
            }
        }
        while(!q.empty()){
            auto t = q.front();
            q.pop();
            for(auto dir:dirs){
                int x = t.first + dir[0];
                int y = t.second + dir[1];
                if(x<0||x>=m||y<0||y>=n||matrix[x][y]<=matrix[t.first][t.second])
                    continue;
                matrix[x][y] = matrix[t.first][t.second]+1;
                q.push({x,y});
            }
        }
        return matrix;
    }
};
class Solution {
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        int m = matrix.size();
        int n = matrix[0].size();
        vector<vector<int>> res(m,vector<int>(n,INT_MAX-1));
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(matrix[i][j]==0)
                    res[i][j] = 0;
                else{
                    if(i>0) res[i][j] = min(res[i][j],res[i-1][j]+1);
                    if(j>0) res[i][j] = min(res[i][j],res[i][j-1]+1);
                }
            }
        }
        for(int i = m-1; i >= 0; i--){
            for(int j = n-1; j >= 0; j--){
                if(res[i][j]!=0&&res[i][j]!=1){
                    if(i < m-1) res[i][j] = min(res[i][j],res[i+1][j]+1);
                    if(j < n-1) res[i][j] = min(res[i][j],res[i][j+1]+1);
                }
            }
        }
        return res;
    }
};