[LeetCode][Java] Spiral Matrix II

来源:互联网 发布:java implements 用法 编辑:程序博客网 时间:2024/06/18 06:49

题目:

Given an integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.

For example,
Given n = 3,

You should return the following matrix:
[ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ]]

题意:

给定一个整数 n,生成一个正方形矩阵。矩阵中包含着从1到n2 这些元素,并且以螺旋顺序排列。

比如,给定n = 3,

你应该返回如下的矩阵:

[ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ]]

算法分析:

和题目《Spiral Matrix》类似,螺旋形的放置,去转就好。

这里需要说明的是最后的情况,《Spiral Matrix》中的矩阵形状不一定是正方形,因此转到最后剩余的元素可能是一个,也可能是一列或者一行。

本题中已经明确说明了生成的方形的矩阵,因此最终的剩余元素只可能为一个,代码中考虑这种情况就OK了。

AC代码:

public class Solution {    public int[][] generateMatrix(int n)     {        int startX = 0;        int startY = 0;        int endX = n - 1;        int endY = n - 1;                int[][] matrix = new int[n][n];        int startValue = 1;                while (startX <= endX)         {            startValue = fillNumber(startX, startY, endX, endY, matrix, startValue);            startX++;//外圈遍历完,遍历内圈            startY++;            endX--;            endY--;        }        return matrix;    }        public int fillNumber(int startX, int startY, int endX, int endY, int[][] matrix, int startV)    {        if (startX == endX)         {            matrix[startX][startY] = startV;            return -1;        }        for (int i = startY; i <= endY; i++) //自左向右        {            matrix[startX][i] = startV;            startV++;        }        for (int i = startX + 1; i <= endX; i++) //自上而下        {            matrix[i][endY] = startV;            startV++;        }        for (int i = endY - 1; i >= startY; i--) //自右而左        {            matrix[endX][i] = startV;            startV++;        }        for (int i = endX - 1; i >= startX + 1; i--)//自下而上        {            matrix[i][startY] = startV;            startV++;        }        return startV;    }}


0 0
原创粉丝点击