LeetCode Exercise 17:01 Matrix
2018-01-12 17:55
176 查看
题名:01矩阵,最近的0
难度:Medium
题目描述:
代码如下:
class Solution {
public:
vector<vector<int>> updateMatrix(vector<vector<int>>& A) {
int r = A.size(), c = A[0].size();
vector<vector<int>> ans(r, vector<int>(c));
queue<pair<int, int>> q;
int D[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
for(int i = 0; i < r; i++){
for(int j = 0; j < c; j++){
if(A[i][j] == 0){
ans[i][j] = 0;
q.push(make_pair(i, j));
}
else{
ans[i][j] = -1;
}
}
}
while(!q.empty()){
pair<int, int> p = q.front();
q.pop();
for(int i = 0; i < 4; i++){
int row = p.first + D[i][0];
int col = p.second + D[i][1];
if(row >= 0 && row < r && col >= 0 && col < c && ans[row][col] == -1){
ans[row][col] = ans[p.first][p.second] + 1;
q.push(make_pair(row, col));
}
}
}
return ans;
}
};
难度:Medium
题目描述:
代码如下:
class Solution {
public:
vector<vector<int>> updateMatrix(vector<vector<int>>& A) {
int r = A.size(), c = A[0].size();
vector<vector<int>> ans(r, vector<int>(c));
queue<pair<int, int>> q;
int D[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
for(int i = 0; i < r; i++){
for(int j = 0; j < c; j++){
if(A[i][j] == 0){
ans[i][j] = 0;
q.push(make_pair(i, j));
}
else{
ans[i][j] = -1;
}
}
}
while(!q.empty()){
pair<int, int> p = q.front();
q.pop();
for(int i = 0; i < 4; i++){
int row = p.first + D[i][0];
int col = p.second + D[i][1];
if(row >= 0 && row < r && col >= 0 && col < c && ans[row][col] == -1){
ans[row][col] = ans[p.first][p.second] + 1;
q.push(make_pair(row, col));
}
}
}
return ans;
}
};
相关文章推荐
- LeetCode Exercise 11: Spiral Matrix
- Spiral Matrix leetcode java
- [LeetCode] Spiral Matrix II
- leetcode -- Set Matrix Zeroes
- leetcode第一刷_Spiral Matrix
- [LeetCode] 74. Search a 2D Matrix 搜索一个二维矩阵
- LeetCode--01 Matrix
- LeetCode | 73. Set Matrix Zeroes
- Leetcode:73. Set Matrix Zeroes
- 【Leetcode】Spiral Matrix
- LeetCode: Set Matrix Zeroes
- 【Leetcode】Spiral Matrix (Rotated)
- [LeetCode] 54. Spiral Matrix
- 【leetcode】Set Matrix Zeroes
- LeetCode - Set Matrix Zeroes
- LeetCode:Search a 2D Matrix
- Leetcode 566 Reshape the Matrix 将数组转换为我们想要的行和列
- [LeetCode] Search a 2D Matrix
- [LeetCode][JavaScript]Search a 2D Matrix
- [LeetCode] 74. Search a 2D Matrix 解题思路