【leetcode】542. 01 Matrix的解法总结

【方法一】
「心路历程」
解这道题的第一印象是bfs，因为类似寻找最短路径。我的第一种解法在不参考任何人的算法的情况下写的，真的是艰辛万苦的打补丁，错了好多testcase，好不容易改出来，尽管写的比较乱，但是运行效率出乎我的意料高。首先将1的地方全赋值为-1且存入队列q。
第一个while用于从前到尾的的先将所有-1的地方根据四周大致赋一遍值，将值大于等于3的记录在栈c中。
第二个while，将所有值大于等于3的点再检查一遍，这里使用栈是为了c从后到前算一遍，因为第一遍是从前到后所以如果前面的值不准确会影响后面的值，所以为了抵消这种影响，c从后面开始。
还有两种特殊情况位于（0，0），因为（0，0）在最开始，没有任何可以参考的值。
总之这个算法不是很规范的运用了某种算法吧，就算是打补丁。
「运行结果」

「代码」

class Solution {
public:
    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        int n=matrix.size();
        int m=matrix[0].size(); 
        //cout<<m;
        typedef pair<int,int>point;
        queue <point> q;
        stack <point> c;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                if(matrix[i][j]==1) 
                {
                    matrix[i][j]=-1;
                    q.push(point(i,j));
                }
            }
        }//n*n 初始化
        point p[4]={point(-1,0),point(0,-1),point(1,0),point(0,1)};

        while(!q.empty())
        {
            int x=q.front().first;
            int y=q.front().second;
            //cout<<x<<y<<endl;
            q.pop();
            for(int i=0;i<4;i++)
            {
                int xx=x+p[i].first;
                int yy=y+p[i].second;
                if(xx>=0 & xx<n & yy>=0 && yy<m)
                {
                    //cout<<xx<<"s"<<yy<<endl;
                    if(matrix[xx][yy]==0) {matrix[x][y]=1;break;}
                    else if(matrix[xx][yy]==-1) 
                    {
                        if(matrix[x][y]<=2)
                            matrix[x][y]=2;
                        else matrix[x][y];
                    }
                    else 
                    {
                        if(matrix[x][y]>=matrix[xx][yy]+1 || matrix[x][y]==-1)
                        {matrix[x][y]=matrix[xx][yy]+1;}
                    }
                    //cout<<matrix[x][y]<<"t"<<endl;
                }
            }
          if(matrix[x][y]>=3){c.push(point(x,y));}
        }
        if(matrix[0][0]==2 && (matrix[0][1]>=2 ||matrix[1][0]>=2))
        {
            matrix[0][0]=1+min(matrix[0][1],matrix[1][0]);
        }
        while(!c.empty())
        {
            int x=c.top().first;
            int y=c.top().second;
            //cout<<x<<" "<<y<<endl;
            int temp=matrix[x][y];
            c.pop();
             for(int i=0;i<4;i++)
            {
                int xx=x+p[i].first;
                int yy=y+p[i].second;
                if(xx>=0 & xx<n & yy>=0 && yy<m)
                {
                   // cout<<xx<<"s"<<yy<<endl;
                    matrix[x][y]=min(matrix[x][y],matrix[xx][yy]+1);
                }
                //cout<<matrix[x][y]<<"t"<<endl;
            }
            if(matrix[x][y]>=3&& matrix[x][y]!=temp)
            {
                c.push(point(x,y));
                //cout<<x<<y<<endl;
             //c.push(point(x,y));
            }
        }
        if(m==1)
        {
            int k=0;
            while(matrix[k][0]!=0)
            {
                k++;
            }
            //cout<<k;
            for(int i=0;i<k;i++)
            {
                matrix[i][0]=k;
                //cout<<matrix[i][0]<<endl;
                k--;
            }
        }
        return matrix;
    }
};

【方法二】
「思路」
用了BFS的方法，跟上面的不同在于，进入队列的是0而不是1。用0的位置来辐射。
「代码」

class Solution {
public:

    vector<vector<int>> updateMatrix(vector<vector<int>>& matrix) {
        int rows = matrix.size();
        int clos = matrix[0].size();
        typedef pair<int, int> point;
        queue<point> q;
        for (int i=0; i<rows; i++) {
            for (int j=0; j<clos; j++) {
                if (matrix[i][j] == 0){
                    q.push(point(i,j));
                }
                else 
                    matrix[i][j] = -1;
            }
        }
        point d[4] = {point(-1,0),point(1,0),point(0,-1),point(0,1)};
        while (q.size()) {
            point temp = q.front();
            q.pop();
            int x = temp.first;
            int y = temp.second;
            for (int i=0; i<4; i++) {
                int xx = x + d[i].first;
                int yy = y + d[i].second;
                if (xx>=0 && xx<rows && yy>=0 && yy<clos && matrix[xx][yy]==-1) {
                    matrix[xx][yy] = matrix[x][y] + 1;
                    q.push(point(xx,yy));
                }
            }
        }
        return matrix;
    }
};

【方法三】
还可以使用dp算法，正在改bug中