[leetcode]Spiral Matrix II

Spiral Matrix II

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*n矩阵中，循环打印1 ~ n^2数字，一圈一圈处理即可，很容易

【注意】：当n为奇数时，最中间的一个数字的处理，我的解法，当n为奇数时，最中间的数字是无法打印的，因此需要特殊处理。

代码如下：

public class Solution {
    public int[][] generateMatrix(int n) {
        if(n <= 0) return new int[0][0];
        int[][] res = new int[n][n];
        int index = 1;
        for(int i = 0; i < n >> 1; i++){
            for(int j = i ; j < n - i; j++){
                res[i][j] = index++;
            }
            for(int j = i + 1; j < n - i; j++){
                res[j][n - 1 - i] = index++;
            }
            for(int j = n - i - 2; j >= i; j--){
                res[n - 1 - i][j] = index++;
            }
            for(int j = n - i - 2; j > i; j-- ){
                res[j][i] = index++;
            }
        }
        if(n % 2 == 1) res[n >> 1][n >> 1] = index;
        return res;
    }
}