标签:遍历 mat int max ret ++ 对角线 leetcode
首先可以确定,每一条对角线其 i + j 的值都是定值,所以可以枚举这个定值;这个定值的范围显然是 [0, n - 1 + m - 1)
然后就是,如果对角线的值为偶数,那么就是从下往上遍历,否则就是从上往下;
对于从上往下的情况,最简单的就是其 i = 0,然后 i 一直递增即可,但显然在后半部分对角线,i 的初始值不是 0,应该是 (i + j) - (m - 1). 但实际上根本不需要区分前后半部分对角线,i 的初始值就是 max(0, (i + j) - (m - 1)).....然后就是 i 的最大值,很显然,后半部分对角线 i 的最大值为 n,而前半部分 i 的最大值为 (i + j).... 即 i 的上限为 min(i + j, n - 1)
对于从下往上的情况与上类似
class Solution { public: vector<int> findDiagonalOrder(vector<vector<int>>& mat) { if(mat.empty() || mat[0].empty()) return {}; int n = mat.size(), m = mat[0].size(); vector<int> ret; ret.reserve(n + m - 2); for(int i = 0; i < n + m - 1; ++ i) { // i 表示 原矩阵 行+列 的值, 因为每一条对角线 行+列 的固定 if(i & 1) { // 奇数, 从上至下 for(int j = max(0, i - (m - 1)); j < min(i + 1, n); ++ j) { // j 表示行 int k = i - j; // k 表示列 ret.emplace_back(mat[j][k]); } } else { // 偶数, 从下至上 for(int k = max(0, i - (n - 1)); k < min(i + 1, m); ++ k) { // k 表示列 int j = i - k; ret.emplace_back(mat[j][k]); } } } return ret; } };
但其实对于第二重 for 循环来说,可以不用算 for 的上界,因为如果 int k = i - j, k < 0 直接 break 即可。
标签:遍历,mat,int,max,ret,++,对角线,leetcode 来源: https://www.cnblogs.com/rookie-acmer/p/15313854.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。