對角線遍歷 (解題記錄)

給定一個含有 M x N 個元素的矩陣(M 行,N 列),請以對角線遍歷的順序返回這個矩陣中的所有元素,對角線遍歷如下圖所示。
示例:

輸入:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
輸出: [1,2,4,7,5,3,6,8,9]

解釋:

在這裏插入圖片描述
說明:

1.給定矩陣中的元素總數不會超過 100000 。

參考大神做法:

vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
	   	vector<int> res;
	    if(matrix.empty()) return res;
	    const int m = matrix.size();
	    const int n = matrix[0].size();
	    for(int l = 0; l < m + n - 1; l++){
	        if(l % 2 == 0){ //左往上
	        	//起始位置 l<m:(1,0) l>m: ((m-1),(l-m+1)) 
	            int i = (l < m ? l : (m - 1));
	            int j = (l < m ? 0 : (l - m + 1));
	            while(i >= 0 && j < n){
	                res.push_back(matrix[i][j]);
	                i--;
	                j++;
	            }
	        }
	        else{
	            int i = (l < n ? 0 : (l - n + 1));
	            int j = (l < n ? l : (n - 1));
	            while(i < m && j >= 0){
	                res.push_back(matrix[i][j]);
	                i++;
	                j--;
	            }
	        }
	    }
	    return res; 
    }

這道題的重點是對邊界條件的判斷。

發佈了28 篇原創文章 · 獲贊 13 · 訪問量 4268
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章