程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> Set Matrix Zeroes

Set Matrix Zeroes

編輯:C++入門知識

題目:


Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

Follow up:
Did you use extra space?
A straight forward solution using O(mn) space is probably a bad idea.
A simple improvement uses O(m + n) space, but still not the best solution.
Could you devise a constant space solution?

分析:用O(mn) space的算法很容易想到,只要再構造一個matrix即可。

用O(m + n) space的算法也比較簡單,只需創建兩個向量,第一個向量記錄哪些行為0,第二個向量記錄哪些列為0即可。

那麼能不能進一步節約空間,答案是顯然的:不需要自己創建兩個向量,而是直接利用矩陣的第一行和第一列,但得先用兩個變量記錄矩陣的第一行和第一列是否為0.


代碼如下:

        void setZeroes(vector<vector<int> > &matrix) {
        int m=matrix.size();
        int n=matrix[0].size();
        bool rowzero=false,columnzero=false;
        for(int i=0;i<m;i++)
        {
            if(matrix[i][0]==0)
            {
                rowzero=true;
                break;
            }
        }
        for(int i=0;i<n;i++)
        {
            if(matrix[0][i]==0)
            {
                columnzero=true;
                break;
            }
        }
        for(int i=1;i<m;i++)
        {
            for(int j=1;j<n;j++)
            {
                if(matrix[i][j]==0)
                {
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }
            }
        }
        for(int i=1;i<m;i++)
        {
            for(int j=1;j<n;j++)
            {
                if(matrix[i][0]==0||matrix[0][j]==0)
                {
                    matrix[i][j]=0;
                }
            }
        }
        if(rowzero)
        {
            for(int i=0;i<m;i++)
            {
                matrix[i][0]=0;
            }
        }
        if(columnzero)
        {
            for(int i=0;i<n;i++)
            {
                matrix[0][i]=0;
            }
        }
    }


 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved