給你一個大小為 m x n 的矩陣 mat ，請以對角線遍歷的順序，用一個數組返回這個矩陣中的所有元素。

代碼思路：以第一行和右邊最后一列作為每輪的開始元素，先用temp存儲，全部按 從左上到右下 的順序遍歷，但是插入到res[ ]數組時，把奇數輪的temp翻轉插入。

class Solution {public int[] findDiagonalOrder(int[][] mat) {if(mat == null||mat.length==0){return new int[0];}int N = mat.length;int M = mat[0].length;int[] res = new int[N*M];int k = 0;ArrayList<Integer> temp = new ArrayList<Integer>();for(int i = 0;i<M+N-1;i++){temp.clear();//確定每輪的開始元素位置int r = i < M ? 0:i-M+1;int c = i<M ?i:M-1;while(r<N&&c>=0){temp.add(mat[r][c]);//temp存儲r++;//從左上到右下c--;//從左上到右下}//將偶數輪翻轉if(i%2==0){Collections.reverse(temp);}//每輪最后都將temp存儲的元素插入，一共M+N輪for(int j = 0;j<temp.size();j++){res[k] = temp.get(j);k++;}} return res;}
}

這題看了答案才看懂，自己一點思路都沒有