LeetCode___59. 螺旋矩阵 II

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

示例:

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

思路：就是按照顺时针的路线填写数字，将这个n*n的矩阵四边分别表示为，l,t,r,b。如图：

先控制其从左向右开始填写，先控制t保持不变，l逐渐递增。for(int i = l; i <= r,i++) arr[t][i] = num++;t++;

然后控制其从上至下开始填写，先控制r保持不变，t主键递增。for(int i = t;i <= b;i++) arr[r][t] = num++;r--;

剩下的和前两步保持一致。

代码：

public static void main(String[] args) {
		int n = 3;
		System.out.println(Arrays.toString(generateMatrix(n)));
	}
	
	
	
	public static int[][] generateMatrix(int n) {
        //定义二维数组
		int[][] arr = new int[n][n];
        //定义矩阵的四个边界，左，右，上，下。
		int l = 0;//左
        int r = n - 1;//右
        int t = 0;//上
        int b = n-1;//下
        
        //定义count，总共有多少个数
        int count = n*n; 
        //定义num,标志进行++
        int num = 1;
        //对矩阵进行顺时针赋值
        while(num <= count){
        	for(int i = l;i <= r;i++) arr[t][i] = num++;
        	t++;
        	for(int i = t;i <= b;i++) arr[i][r] = num++;
        	r--;
        	for(int i = r;i >= l;i--) arr[b][i] = num++;
        	b--;
        	for(int i = b;i >= t;i--) arr[i][l] = num++;
        	l++;
        }
        
        
        return arr;
    }