队列&栈//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
注意:
- 给定矩阵的元素个数不超过 10000。
- 给定矩阵中至少有一个元素是 0。
- 矩阵中的元素只在四个方向上相邻: 上、下、左、右。
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;
}
};
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· AI与.NET技术实操系列:基于图像分类模型对图像进行分类
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 25岁的心里话
· 闲置电脑爆改个人服务器(超详细) #公网映射 #Vmware虚拟网络编辑器
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· 零经验选手,Compose 一天开发一款小游戏!
· 一起来玩mcp_server_sqlite,让AI帮你做增删改查!!