码迷,mamicode.com
首页 > 其他好文 > 详细

矩阵置零

时间:2018-07-15 16:16:37      阅读:312      评论:0      收藏:0      [点我收藏+]

标签:i++   代码   style   baidu   get   使用   开始   ++   http   

 

给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用原地算法。

示例 1:

输入: 
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出: 
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

示例 2:

输入: 
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
输出: 
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

一开始我想的是找到所有带0的行和列 放到row[m]和col[n]里,这样需要m+n的额外空间。
后来看了别人的解后 发现可以把第一行第一列当做上面的row[m]和col[n]来用,当然这就需要先记录下首行首列是否要置0,这样只需要2个int的额外空间。
下面是是代码,写得比较散,但是便于理解,以后可以把它们合并起来。
void setZeroes(int** matrix, int matrixRowSize, int matrixColSize) {
    if(matrixRowSize==0||matrixColSize==0)
        return;
    int m=matrixRowSize;
    int n=matrixColSize;
    int i,j;
    int firstRow=0;
    int firstCol=0;
    for(i=0;i<m;i++)
        if(matrix[i][0]==0)
        {
            firstCol=1;
            break;
        }
    for(j=0;j<n;j++)
        if(matrix[0][j]==0)
        {
            firstRow=1;
            break;
        }
    for(i=1;i<m;i++)
    {
        for(j=1;j<n;j++)
            if(matrix[i][j]==0)
            {
                matrix[0][j]=0;
                matrix[i][0]=0;
            }
    }
    for(i=1;i<m;i++)
    {
        if(matrix[i][0]==0)
        {
            for(j=1;j<n;j++)
                matrix[i][j]=0;
        }
    }
    for(j=1;j<n;j++)
    {
        if(matrix[0][j]==0)
        {
            for(i=1;i<m;i++)
                matrix[i][j]=0;
        }
    }
    if(firstRow)
    {
        for(j=0;j<n;j++)
            matrix[0][j]=0;
    }
    if(firstCol)
    {
        for(i=0;i<m;i++)
            matrix[i][0]=0;
    }
}

矩阵置零

标签:i++   代码   style   baidu   get   使用   开始   ++   http   

原文地址:https://www.cnblogs.com/onlyandonly/p/9313515.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!