您好,登錄后才能下訂單哦!
今天小編給大家分享一下C++如何實(shí)現(xiàn)矩陣賦零的相關(guān)知識點(diǎn),內(nèi)容詳細(xì),邏輯清晰,相信大部分人都還太了解這方面的知識,所以分享這篇文章給大家參考一下,希望大家閱讀完這篇文章后有所收獲,下面我們一起來了解一下吧。
Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.
click to show follow up.
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?
據(jù)說這題是CareerCup上的原題,我還沒有刷CareerCup,所以不知道啦,不過這題也不算難,雖然我也是看了網(wǎng)上的解法照著寫的,但是下次遇到絕對想的起來。這道題中說的空間復(fù)雜度為O(mn)的解法自不用多說,直接新建一個(gè)和matrix等大小的矩陣,然后一行一行的掃,只要有0,就將新建的矩陣的對應(yīng)行全賦0,行掃完再掃列,然后把更新完的矩陣賦給matrix即可,這個(gè)算法的空間復(fù)雜度太高。將其優(yōu)化到O(m+n)的方法是,用一個(gè)長度為m的一維數(shù)組記錄各行中是否有0,用一個(gè)長度為n的一維數(shù)組記錄各列中是否有0,最后直接更新matrix數(shù)組即可。這道題的要求是用O(1)的空間,那么我們就不能新建數(shù)組,我們考慮就用原數(shù)組的第一行第一列來記錄各行各列是否有0.
- 先掃描第一行第一列,如果有0,則將各自的flag設(shè)置為true
- 然后掃描除去第一行第一列的整個(gè)數(shù)組,如果有0,則將對應(yīng)的第一行和第一列的數(shù)字賦0
- 再次遍歷除去第一行第一列的整個(gè)數(shù)組,如果對應(yīng)的第一行和第一列的數(shù)字有一個(gè)為0,則將當(dāng)前值賦0
- 最后根據(jù)第一行第一列的flag來更新第一行第一列
代碼如下:
class Solution { public: void setZeroes(vector<vector<int> > &matrix) { if (matrix.empty() || matrix[0].empty()) return; int m = matrix.size(), n = matrix[0].size(); bool rowZero = false, colZero = false; for (int i = 0; i < m; ++i) { if (matrix[i][0] == 0) colZero = true; } for (int i = 0; i < n; ++i) { if (matrix[0][i] == 0) rowZero = true; } for (int i = 1; i < m; ++i) { for (int j = 1; j < n; ++j) { if (matrix[i][j] == 0) { matrix[0][j] = 0; matrix[i][0] = 0; } } } for (int i = 1; i < m; ++i) { for (int j = 1; j < n; ++j) { if (matrix[0][j] == 0 || matrix[i][0] == 0) { matrix[i][j] = 0; } } } if (rowZero) { for (int i = 0; i < n; ++i) matrix[0][i] = 0; } if (colZero) { for (int i = 0; i < m; ++i) matrix[i][0] = 0; } } };
以上就是“C++如何實(shí)現(xiàn)矩陣賦零”這篇文章的所有內(nèi)容,感謝各位的閱讀!相信大家閱讀完這篇文章都有很大的收獲,小編每天都會為大家更新不同的知識,如果還想學(xué)習(xí)更多的知識,請關(guān)注億速云行業(yè)資訊頻道。
免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。