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 ] ]
题解:以前做过的Spiral Matrix是给一个矩阵螺旋式的输出,这道题是给一个n,螺旋式的生成矩阵。和第一个的做法类似,定义四个变量
row_left:当前未填的子矩阵的最左边界
row_right:当前未填的子矩阵的最右边界
column_up:当前未填子矩阵的最上边界
column_down:当前未填子矩阵的最下边界
然后每次进行四个循环,做一个螺旋。
最后一个元素要单独处理(其实只要n是奇数的时候单独处理),因为这时只有一个元素未填,不够运行一个螺旋,如果还是用四个for循环就会重复修改已经放置好的元素。
代码如下:
1 public class Solution { 2 public int[][] generateMatrix(int n) { 3 int matrix[][] = new int[n][n]; 4 int row_up = 0; 5 int row_down = n - 1; 6 int column_left = 0; 7 int column_right = n - 1; 8 int number = 1; 9 10 while(number <= n*n-1){ 11 for(int i = column_left;i <= column_right;i++){ 12 matrix[row_up][i] = number; 13 number++; 14 } 15 row_up++; 16 17 for(int j = row_up;j <= row_down;j++){ 18 matrix[j][column_right] = number; 19 number++; 20 } 21 column_right--; 22 23 for(int j = column_right;j >= column_left;j --){ 24 matrix[row_down][j] = number; 25 number++; 26 } 27 row_down--; 28 29 for(int i = row_down;i >= row_up;i --){ 30 matrix[i][column_left] = number; 31 number++; 32 } 33 column_left++; 34 35 } 36 37 if(n > 0) 38 matrix[n/2][n%2==0?n/2-1:n/2] = n*n; 39 return matrix; 40 } 41 }
还有个坑就是n等于0的时候,就没有所谓的“最后一个元素单独处理了”,这中情况要判断,否则会抛出异常。
【leetcode刷题笔记】Spiral Matrix II,布布扣,bubuko.com
【leetcode刷题笔记】Spiral Matrix II
原文地址:http://www.cnblogs.com/sunshineatnoon/p/3830262.html