leetcode.498 對角線遍歷

給定一個含有 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 。

 

思路：設置一個標誌位來控制方向的反轉，以及用數組來控制方向的走向，然後就是各種判斷邊界條件了。

代碼：

class Solution {
    public int[] findDiagonalOrder(int[][] matrix) {
       if(matrix.length == 0 || matrix[0].length == 0) return new int[0];
        
        int m = matrix.length;    // 行數
        int n = matrix[0].length;    // 列數
        int[] order = new int[m * n];   // 結果數
        
        int row = 0;
        int col = 0;
        int k = 0;   // 控制方向的轉變
        int[][] dir = {{-1,1},{1,-1}};    // 右上，左下的方向
        for(int i = 0;i < order.length;i++){
            order[i] = matrix[row][col];
            row += dir[k][0];
            col += dir[k][1];
            // 判斷越界情況，並分清反轉方向 
            if(col > n-1){
                row += 2;
                col = n-1;
                k = 1-k;
            }
            if(row < 0){
                row = 0;
                k = 1-k;
            }
            
            if(row > m-1){
                row = m-1;
                col += 2;
                k = 1-k;
            }
            
            if(col < 0){
                col = 0;
                k = 1-k;
            }
           
            
        }
        return order;
    }
}