<LeetCode>542. 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. 矩阵中的元素只在四个方向上相邻: 上、下、左、右

解题思路——

1. 一定要明确下面的数据结构方面的内容，pair<int,int>也就是tp，放的是图中元素的下标<i,j>。而vector<vector<int>>

res是一张图，这张图作为返回值，图中元素是0和正数，正数代表了距离最近的0的距离。

2. BFS的思想是，找到0和0的下标<i,j>，然后以这个0为中心进行广度搜索，每次搜索之后入队。

3. 顺承2的思想，显示以0为中心，然后队列里的0的下标的那些（一定要牢记，队列里放的是下标）搜索完之后，就是1的下标们，然后2的下标们，直到全图被搜索完毕。

4. 一定要弄清楚第一点啊！

class Solution {
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        //第一步，目标图res将原来的所有0保留，将原来的所有1设置为-1
        vector<vector<int>> res = matrix;
        int m= matrix.size(), n = matrix[0].size();
        if(m<=0||n<=0)  return res;
        //对目标矩阵进行处理
        typedef pair<int,int> tp;//这个数据结构放的是二维数组的下标
        queue<tp> q;
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
            {
                if(matrix[i][j] == 0)
                    q.push(tp(i,j));//记住，res是张图，这个操作保证图里的0不变，但是，把下标入队列
                else
                    res[i][j] = -1;//相当于标志位，为什么不放1呢？因为在res这个图里，正数代表距离，防止1的二义性
            }        
        //第二步，设置方向
        vector<tp> dir = {tp(-1,0),tp(1,0),tp(0,-1),tp(0,1)};
        //第三步，BFS
        while(!q.empty())
        {
            tp front = q.front();
            q.pop();
            int x = front.first,y = front.second;
            for(int i=0;i<4;i++)//对四个方向进行bfs
            {
                int nx=x+dir[i].first,ny=y+dir[i].second;
                if(nx<m&&nx>=0&&ny<n&&ny>=0&&res[nx][ny]==-1)
                {
                    res[nx][ny] = res[x][y] + 1;
                    q.push(tp(nx,ny));
                }
            }
        }

        return res;
    }
};