1003. 最近的0

Problem:

输入一个N*M的01矩阵A，对矩阵的每个位置，求至少经过多少步可以到达一个0. 每一步可以往上下左右走一格.

 

请为下面的Solution类实现解决这一问题的函数nearestZero，函数参数A为给出的01矩阵，A的行数和列数均不大于100. 函数的返回值是问题的答案.

 

class Solution {

public:

    vector<vector<int>> nearestZero(vector<vector<int>>& A) {

                        

    }

};

 

例如：

A=

1 1 1

0 1 1

0 0 1

 

答案为

1 2 3

0 1 2

0 0 1

 

注意：你只需要提交Solution类的代码，你在本地可以编写main函数测试程序，但不需要提交main函数的代码. 注意不要修改类和函数的名称.

Solution:

#include<algorithm>
#include<stdio.h>
#include<iostream>
#include<cstring>
#include<stdlib.h>
#include<vector>


using namespace std;


class Solution {
public:
vector<vector<int>> nearestZero(vector<vector<int>>& A) {
vector<vector<int>> b;
b.resize(A.size());
for (int i = 0; i < A.size(); i++) b[i].resize(A[i].size());//给定空间

for (int i = 0; i < A.size(); i++)//初始化
{
for (int j = 0; j < A[i].size(); j++)
{
if (A[i][j] == 0)
{
b[i][j] = 0;
}
else
{
b[i][j] = -1;
}
}
}


bool flag = true;
int num = 0;
while (flag)
{
flag = false;
for (int i = 0; i < b.size(); i++)
{
for (int j = 0; j < b[i].size(); j++)
{
if (b[i][j] == num)
{
flag = true;
if ((i > 0) && (b[i - 1][j] == -1))
{
b[i - 1][j] = num + 1;
}
if ((i < b.size() - 1) && (b[i + 1][j] == -1))
{
b[i + 1][j] = num + 1;
}
if ((j > 0) && (b[i][j - 1] == -1))
{
b[i][j - 1] = num + 1;
}
if ((j < b[i].size() - 1) && (b[i][j + 1] == -1))
{
b[i][j + 1] = num + 1;
}
}
}
}
num++;
}
return b;
}
};