LeetCode 59 螺旋矩阵II (Java)

301 阅读1分钟

LeetCode 59 螺旋矩阵II

题目描述:

给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]

img

代码:

class Solution {
  public int[][] generateMatrix(int n) {
    int[][] matrix = new int[n][n];
    int top = 0;
    int left = 0;
    int right = matrix.length-1; //获取列数
    int bottom = matrix[0].length-1; 	//获取行数
    int count = 1;
		while(true) {
				for(int i = left;i <= right;i++) matrix[top][i] = count++;
				top++;
				if(left > right || top > bottom) break;
				
				for(int i = top;i <= bottom;i++) matrix[i][right] = count++;
				right--;
				if(left > right || top > bottom) break;
				
				for(int i = right;i >= left;i--) matrix[bottom][i] = count++;
				bottom--;
				if(left > right || top > bottom) break;
				
				for(int i = bottom;i >= top;i--) matrix[i][left] = count++;
				left++;
				if(left > right || top>bottom) break;
		}
		return matrix;
   }
}

image.png