力扣：498. 对角线遍历

what is an object

于 2022-06-14 22:35:50 发布

阅读量65

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签： leetcode java 算法

本文链接：https://blog.youkuaiyun.com/newOneObject/article/details/125287613

算法专栏收录该内容

278 篇文章

订阅专栏

本文档介绍了如何通过Java实现一个解决方案类，解决将二维矩阵按对角线顺序排列的问题。通过`findDiagonalOrder`方法，处理不同大小的矩阵并返回按顺序的数组。核心是巧妙地利用奇偶数判断行和列的遍历方向。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >



import java.util.Arrays;

/**
 * @author xnl
 * @Description:
 * @date: 2022/6/14   21:09
 */
public class Solution {
    public static void main(String[] args) {
        Solution solution = new Solution();
        // int[][] mat = {{1,2,3},{4,5,6},{7,8,9}};
        // int[][] mat = {{2, 3}};
        int[][] mat = {{3}, {2}};
        System.out.println(Arrays.toString(solution.findDiagonalOrder(mat)));
    }

    /**
     * @param mat
     * @return
     */
    public int[] findDiagonalOrder(int[][] mat) {
        if (mat.length == 1){
            return mat[0];
        }
        int m = mat.length;
        int n = mat[0].length;
        // 一共有的边
        int edge = m + n - 1;
        int[] ans = new int[m * n];
        int index = 0;

        for (int i = 0; i < edge; i++){
            // 奇数代表从上到下，偶数代表从下到上
            if (i % 2 == 1){
                int x = i < n ? 0 : i - n + 1;
                int y = i < n ? i : n - 1;
                while (x < m && y >= 0){
                    ans[index] = mat[x][y];
                    index++;
                    x++;
                    y--;
                }
            } else {
                int x = i < m ? i : m - 1;
                int y = i < m ? 0 : i - m + 1;
                while (x >= 0 && y < n){
                    ans[index] = mat[x][y];
                    index++;
                    x--;
                    y++;
                }
            }
        }
        return ans;
    }
}