leetcode 59 螺旋矩阵 II

给定一个正整数 n，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的正方形矩阵。

示例:

输入: 3
输出:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

 

class Solution {
public int[][] generateMatrix(int n) {
        int count = 1;
        int [][]num = new int[n][n];//注意此处应该用n而不是一个无限大的数
        int row = 0,col = 0;
        num[0][0] = 1;//第一步一定正确，只需要判断下一步。而不能判断当前的值！
        while(count < n * n){
            //&&为短路运算符，若前面不满足，则后面一定不会计算
            //注意数组的赋值应放在后方！！！
            while(col < n - 1 && num[row][col + 1] == 0) {
                num[row][++col] = ++count;
            }
            while(row < n - 1 && num[row + 1][col] == 0) {
                num[++row][col] = ++count;
            }
            while(col > 0 && num[row][col - 1] == 0){
                num[row][--col] = ++count;
            }
            while(row > 0 && num[row - 1][col] == 0){
                num[--row][col] = ++count;
            }
        }
        return num;
    }
}