码迷,mamicode.com
首页 > 其他好文 > 详细

【leetcode刷题笔记】Spiral Matrix II

时间:2014-07-08 23:54:39      阅读:407      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   for   io   代码   

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

标签:style   blog   color   for   io   代码   

原文地址:http://www.cnblogs.com/sunshineatnoon/p/3830262.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!