【leetcode热题100】螺旋矩阵 II

最新推荐文章于 2024-10-01 05:04:32 发布

kiugvui

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量359

点赞数 8

分类专栏： leetcode热题文章标签： leetcode 矩阵算法

本文链接：https://blog.csdn.net/kiugvui/article/details/135796112

版权

leetcode热题专栏收录该内容

170 篇文章 1 订阅

订阅专栏

给你一个正整数 n ，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

示例 1：

输入：n = 3
输出：[[1,2,3],[8,9,4],[7,6,5]]

示例 2：

输入：n = 1
输出：[[1]]

解法一

直接按照 54题，贪吃蛇的走法来写，如果没做过可以看一下。

/*
     * direction 0 代表向右, 1 代表向下, 2 代表向左, 3 代表向上
     */
public int[][] generateMatrix(int n) {
    int[][] ans = new int[n][n];
    int start_x = 0, start_y = 0, direction = 0, top_border = -1, // 上边界
    right_border = n, // 右边界
    bottom_border = n, // 下边界
    left_border = -1; // 左边界
    int count = 1;
    while (true) {
        // 全部遍历完结束
        if (count == n * n + 1) {
            return ans;
        }
        // 注意 y 方向写在前边，x 方向写在后边
        ans[start_y][start_x] = count;
        count++;
        switch (direction) {
                // 当前向右
            case 0:
                // 继续向右是否到达边界
                // 到达边界就改变方向，并且更新上边界
                if (start_x + 1 == right_border) {
                    direction = 1;
                    start_y += 1;
                    top_border += 1;
                } else {
                    start_x += 1;
                }
                break;
                // 当前向下
            case 1:
                // 继续向下是否到达边界
                // 到达边界就改变方向，并且更新右边界
                if (start_y + 1 == bottom_border) {
                    direction = 2;
                    start_x -= 1;
                    right_border -= 1;
                } else {
                    start_y += 1;
                }
                break;
            case 2:
                if (start_x - 1 == left_border) {
                    direction = 3;
                    start_y -= 1;
                    bottom_border -= 1;
                } else {
                    start_x -= 1;
                }
                break;
            case 3:
                if (start_y - 1 == top_border) {
                    direction = 0;
                    start_x += 1;
                    left_border += 1;
                } else {
                    start_y -= 1;
                }
                break;
        }
    }

}

时间复杂度：O（n²）。

空间复杂度：O（1）。