LeetCode 498题对角线遍历

、sakura

已于 2022-03-28 23:32:09 修改

阅读量2.7k

点赞数

文章标签： leetcode 算法职场和发展

于 2022-01-26 16:56:12 首次发布

本文链接：https://blog.csdn.net/sakuraSD/article/details/122705206

版权

LeetCode 498题
这个代码也写的挺优美的。哈哈
在这里插入图片描述

class Solution {
    public int[] findDiagonalOrder(int[][] mat) {
        int flag = 0;//flag偶数，向上遍历，奇数则向下遍历
        int col = mat.length;//行
        int row = mat[0].length;//列
        int count = 0;//计数，count等于mat的元素个数，则遍历结束
        int[] resArr = new int[col*row];
        int x = 0;
        int y = 0;
		while(count < col * row) {
            if(flag % 2 == 0) {//向上遍历
                if(flag < col) {
                    x = flag;
                    y = flag - x;
                }else {
                    x = col - 1;
                    y = flag - x;
                }
				for(;x>-1&&x<col&&y>-1&&y<row;) {
                    resArr[count] = mat[x][y];
                    x--;
                    y++;
                    count++;
                }
                flag++;
            }else {//向下遍历
			if(flag < row) {
                    y = flag;
                    x = flag - y;
                }else {
                    y = row - 1;
                    x = flag - y;
                }
                for(;x>-1&&x<col&&y>-1&&y<row;) {
                    resArr[count] = mat[x][y];
                    x++;
                    y--;
                    count++;
                }
                flag++;
				}
        }
        return resArr;
    }
}