leetcode:螺旋矩阵II(java)

package LeetCode;
/*
给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]
 */
public class GenerateMatrix {
    public int[][] generateMatrix(int n) {
        int[][] res=new int[n][n];
        if (n == 0) {
            return res;
        }
        int left=0;
        int right=n-1;
        int up=0;
        int down=n-1;
        int i=1;
       while(i<=n*n){
           //最上面的从最左边到最右边 横
           for (int col = left; col <= right; col++) {
               res[up][col]=i;
               i++;
           }
           //最上面的一行已经加完了 上边向下走一行所以++
           up++;
           if (i<=n*n){
               for (int j = up; j <=down ; j++) {
                   res[j][right]=i;
                   i++;
               }
               right--;
           }
           if (i<=n*n){
               for (int j = right; j >=left ; j--) {
                   res[down][j]=i;
                   i++;
               }
               down--;
           }
           if (i<=n*n){
               for (int j = down; j >=up ; j--) {
                   res[j][left]=i;
                   i++;
               }
               left++;
           }
       }
        return res;
    }

    public static void main(String[] args) {
        GenerateMatrix a=new GenerateMatrix();
        a.generateMatrix(3);
    }
}

 

你可能感兴趣的:(算法试题(java))