Leetcode 59. 螺旋打印矩阵

题目描述

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

示例 1：
输入：n = 3
输出：[[1,2,3],[8,9,4],[7,6,5]]
示例 2：

输入：n = 1
输出：[[1]]

解题思路

此题相对上一题leetcode54更简单，因为边界更可控，是正方形，可以从外圈向中央逼近，重点是边界的处理，避免重复

代码实现

class Solution {
    public int[][] generateMatrix(int n) {
        int[][] r = new int[n][n];
        int lr = 0, rr = n -1, sum = n* n;
        fill(r, lr, rr, 1, sum);
        return r;
    }

    private void fill(int[][] r, int lr, int rr, int n, int sum) {
       if(n > sum) return;
        if(lr > rr) return;
        for (int i = lr; i <= rr; i++) {
            r[lr][i] = n;
            n++;
        }
        for (int i = lr+1; i <= rr; i++) {
            r[i][rr] = n;
            n++;
        }
        for (int i = rr-1; i >= lr; i--) {
            r[rr][i] = n;
            n++;
        }
        for (int i = rr-1; i > lr; i--) {
            r[i][lr] = n;
            n++;
        }
        fill(r, lr+1, rr-1, n, sum);
    }
}