题目描述

给定一个整数n，将数字1到n2n^2n2按螺旋的顺序填入n×n的矩阵
例如：给出的n=3，你应该返回如下矩阵： 
[↵ [ 1, 2, 3 ],↵ [ 8, 9, 4 ],↵ [ 7, 6, 5 ]↵]
代码实现： 
import java.util.*;


public class Solution {
    /**
     * 
     * @param n int整型 
     * @return int整型二维数组
     */
    public int[][] generateMatrix (int n) {
        // write code here
        int[][] res=new int[n][n];
        if(n<1)
            return res;
        int index=1,rowStart=0,rowEnd=n-1,colStart=0,colEnd=n-1;    //设置参数规定每次循环的首与尾
        while(index<=n*n){
            for(int i=colStart;i<=colEnd;i++){                  //→
                res[rowStart][i]=index++;
            }
            for(int i=rowStart+1;i<=rowEnd;i++){                //↓
                res[i][colEnd]=index++;
            }
            for(int i=colEnd-1;i>=colStart;i--){               //←
                res[rowEnd][i]=index++;
            }
            for(int i=rowEnd-1;i>rowStart;i--){                //↑
                res[i][colStart]=index++;
            }
            rowStart++;
            rowEnd--;
            colStart++;
            colEnd--;
        }
        return res;
    }
}