leetcode498 对角线遍历

给定一个含有 M x N 个元素的矩阵（M 行，N 列），请以对角线遍历的顺序返回这个矩阵中的所有元素，对角线遍历如下图所示。

示例:

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

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

解释:

说明:

给定矩阵中的元素总数不会超过 100000 。

思路：想法是对角线行+列从0到m+n-2变化，奇数和偶数方向不同，但是写出的代码效率不高

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
        vector<int>res;
        if(matrix.size()==0 || matrix[0].size()==0) return res;
        int m = matrix.size(),n = matrix[0].size();
        
        for(int k=0;k<m+n-1;k++){
            if(k%2 == 0){
                int i = k,j = 0;
                while(j<n && j<=k){
                    if(i<m)
                        res.push_back(matrix[i][j]);
                    j++;
                    i = k-j;
                } 
            }
            else{
                int i = 0,j = k-i;
                while(i<m && i<=k){
                    if(j<n)
                        res.push_back(matrix[i][j]);
                    i++;
                    j=k-i;
                } 
            }
        }
        return res;
    }
};

下面是思路比较巧妙的leetcode解法：

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
        vector<int>res;
        if(matrix.size()==0 || matrix[0].size()==0) return res;
        int m = matrix.size(),n = matrix[0].size();
        int i = 0,j = 0;
        for(int k=0;k<m*n;k++){
            res.push_back(matrix[i][j]);
            if((i+j)%2 == 0){
                if(j == n-1){
                    i++;
                }else if(i == 0){
                    j++;
                }else{
                    i--;
                    j++;
                }
                 
            }
            else{
                if(i == m-1){
                    j++;
                }else if(j == 0){
                    i++;
                }else{
                    i++;
                    j--;
                }
            }
        }
        return res;
    }
};