59. 螺旋矩阵 II

//给你一个正整数 n ，生成一个包含 1 到 n² 所有元素，且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。 
//
// 
//
// 示例 1： 
//
// 
//输入：n = 3
//输出：[[1,2,3],[8,9,4],[7,6,5]]
// 
//
// 示例 2： 
//
// 
//输入：n = 1
//输出：[[1]]
// 
//
// 
//
// 提示： 
//
// 
// 1 <= n <= 20 
// 
// Related Topics 数组 矩阵 模拟 👍 733 👎 0

import java.util.Arrays;

//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public int[][] generateMatrix(int n) {
        int[][] arr = new int[n][n];
        int index = 1;
        int i = 0;
        int j = 0;

        while (true) {
            int flag = 0;
            while (j >= 0 && j < n && arr[i][j] == 0) {
                arr[i][j++] = index++;
                flag = 1;
            }

            j--;
            i++;
            while (i >= 0 && i < n && arr[i][j] == 0) {
                arr[i++][j] = index++;
                if (flag == 1)
                    flag = 2;
            }

            i--;
            j--;
            while (n > j && j >= 0 && arr[i][j] == 0) {
                arr[i][j--] = index++;
                if (flag == 2)
                    flag = 3;
            }

            j++;
            i--;

            while (n > i && i >= 0 && arr[i][j] == 0) {
                arr[i--][j] = index++;
                if (flag == 3)
                    flag = 4;
            }
            i++;
            j++;
            if (flag != 4)
                break;
        }
        return arr;
    }
}
//leetcode submit region end(Prohibit modification and deletion)