Leetcode 498. 对角线遍历 [硬模拟]

最新推荐文章于 2024-09-18 20:54:48 发布

9ack!?

最新推荐文章于 2024-09-18 20:54:48 发布

阅读量1.3k

点赞数 2

分类专栏：题解

本文链接：https://blog.csdn.net/apple_52296856/article/details/125280561

版权

矩阵遍历算法对角线数据结构算法实现

关键词由CSDN通过智能技术生成

题解专栏收录该内容

39 篇文章 0 订阅

订阅专栏

题目

思路

画个图找一下规律，简单模拟一下就出来了。

代码

class Solution {
    public:
        vector<int> findDiagonalOrder(vector<vector<int>>& mat) {
            vector<int> vec;
            int i, j;
            i = j = 0;
            bool flag = true;
            while(flag) {
                while(flag) {
                    if(i >= mat.size() || j >= mat[0].size()) {flag = false; break;}
                    vec.push_back(mat[i][j]);
                    i -= 1; j += 1;
                    if(j >= mat[0].size()) {
                        j -= 1;
                        i += 2;
                        flag = false;
                        break;
                    } else if(i < 0) {
                        i += 1;
                        flag = false;
                        break;
                    }
                }
                while(!flag) {
                    if(i >= mat.size() || j >= mat[0].size()) {flag = false; break;}
                    vec.push_back(mat[i][j]);
                    j -= 1; i += 1;
                    if(i >= mat.size()) {
                        i -= 1;
                        j += 2;
                        flag = true;
                        break;
                    } else if(j < 0) {
                        j += 1;
                        flag = true;
                        break;
                    }
                }
            }
            return vec;
        }
};