(LeetCode 每日一题) 498. 对角线遍历 (矩阵、模拟)

题目：498. 对角线遍历

思路：矩阵模拟，时间复杂度0(nm)。
按对角线遍历，而每条对角线都是i+j=k，k的范围为0,n-1+m-1。

C++版本：

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& mat) {
        int n=mat.size(),m=mat[0].size();
        vector<int> ans;
        // i+j=k  -> j=k-i;
        for(int k=0;k<n+m-1;k++){
        	//枚举j的值，那需要求出j可达的最小值和最大值
        	//当i=n-1时，j的值最小，但不能小于0
            int mn=max(k-(n-1),0);
            //当i=0时，j的值最大，但不能大于m-1
            int mx=min(k-0,m-1);
            if(k%2==0){
                for(int j=mn;j<=mx;j++){
                    ans.push_back(mat[k-j][j]);
                }
            }else{
                for(int j=mx;j>=mn;j--){
                    ans.push_back(mat[k-j][j]);
                }
            }
        }
        return ans;
    }
};

JAVA版本：

class Solution {
    public int[] findDiagonalOrder(int[][] mat) {
        int n=mat.length,m=mat[0].length;
        int[] ans=new int[n*m];
        int idx=0;
        // i+j=k  -> j=k-i;
        for(int k=0;k<n+m-1;k++){
            int mn=Math.max(k-(n-1),0);
            int mx=Math.min(k-0,m-1);
            if(k%2==0){
                for(int j=mn;j<=mx;j++){
                    ans[idx++]=mat[k-j][j];
                }
            }else{
                for(int j=mx;j>=mn;j--){
                    ans[idx++]=mat[k-j][j];
                }
            }
        }
        return ans;
    }
}

GO版本：

func findDiagonalOrder(mat [][]int) []int {
    n,m:=len(mat),len(mat[0])
    ans:=make([]int,n*m)
    idx:=0
    // i+j=k  -> j=k-i;
    for k:=0;k<n+m-1;k++ {
        mn:=max(k-(n-1),0)
        mx:=min(k-0,m-1)
        if k%2==0 {
            for j:=mn;j<=mx;j++ {
                ans[idx]=mat[k-j][j]
                idx++
            }
        }else{
            for j:=mx;j>=mn;j-- {
                ans[idx]=mat[k-j][j]
                idx++
            }
        }
    }
    return ans
}