1. 程式人生 > >Leetcode __59. 螺旋矩陣 II

Leetcode __59. 螺旋矩陣 II

問題描述

給定一個正整數 n,生成一個包含 1 到 n2 所有元素,且元素按順時針順序螺旋排列的正方形矩陣。

示例:

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

解題思路

不要怕碼放,一圈一圈轉,一個邊一個邊遍歷 在這裡插入圖片描述

在這裡插入圖片描述

邊界值為閉區間

實現

class Solution {
    public int[][] generateMatrix(int n) {
        if(n==1){
            int[][] data=new int[1][1];
            data[0][0]=1;
            return data;
        }
        int times = n % 2 == 0 ? n / 2 : n / 2 + 1;
        int c1 = 0;
        int c2 = n - 1;
        int r1 = 0;
        int r2 = n - 1;
        int begin = 1;
        int[][] res = new int[n][n];
        for (int i = 0; i < times; i++) {
            for (int c = c1; c <= c2; c++) {
                res[r1][c] = begin++;
            }
            for (int j = r1 + 1; j <= r2; j++) {
                res[j][r2] = begin++;
            }
            if (c1 < c2 && r1 < r2) {
                for (int a = c2 - 1; a >=c1+1; a--) {
                    res[r2][a] = begin++;
                }
                for (int b = r2; b >=r1+1 ; b--) {
                    res[b][c1] = begin++;
                }
            }
            c1++;
            c2--;
            r1++;
            r2--;
        }
        return res;

    }
}