leetcode:59. 螺旋矩阵 II

一时失言乱红尘 2022-04-05 08:50 213阅读 0赞

## 题目: ##

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

示例:

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

## 分析: ##

螺旋旋转往数组里添加数字即可

## 代码: ##

public int[][] generateMatrix(int n) {
    		int[][] matrix = new int[n][n];
    		if (n == 0)
    			return matrix;
    
    		// 记录当前方向,0,1,2,3表示向右,向下,向左,向上
    		int direction = 0;
    		// 记录每个方向前进次数.0,1,2,3记录右,下,左,上
    		int[] counter = new int[4];
    
    		loop(matrix, direction, counter, 0, 0, 1);
    		return matrix;
    	}
    
    	private void loop(int[][] matrix, int direction, int[] counter, int i, int j, int count) {
    		matrix[i][j] = count;
    		// 螺旋完毕
    		if (count == matrix.length * matrix[0].length) {
    			return;
    		}
    		// 先检测当前方向,再查看是否需要转向
    		switch (direction) {
    		// 向右
    		case 0:
    			// 需要转向了
    			if (j + 1 + counter[1] == matrix[0].length) {
    				counter[0] += 1;
    				direction = 1;
    				loop(matrix, direction, counter, i + 1, j, count + 1);
    				return;
    			}
    			// 不需要转向
    			loop(matrix, direction, counter, i, j + 1, count + 1);
    			break;
    		// 向下
    		case 1:
    			// 需要转向了
    			if (i + 1 + counter[2] == matrix.length) {
    				counter[1] += 1;
    				direction = 2;
    				loop(matrix, direction, counter, i, j - 1, count + 1);
    				return;
    			}
    			// 不需要转向
    			loop(matrix, direction, counter, i + 1, j, count + 1);
    			break;
    		// 向左
    		case 2:
    			// 需要转向了
    			if (j == counter[3]) {
    				counter[2] += 1;
    				direction = 3;
    				loop(matrix, direction, counter, i - 1, j, count + 1);
    				return;
    			}
    			// 不需要转向
    			loop(matrix, direction, counter, i, j - 1, count + 1);
    			break;
    		// 向上
    		case 3:
    			// 需要转向了
    			if (i == counter[0]) {
    				counter[3] += 1;
    				direction = 0;
    				loop(matrix, direction, counter, i, j + 1, count + 1);
    				return;
    			}
    			// 不需要转向
    			loop(matrix, direction, counter, i - 1, j, count + 1);
    			break;
    		}
    	}

## 效率: ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI0NTE2NTQ5_size_16_color_FFFFFF_t_70]

## 总结: ##

效率还可以,这里因为没转到的直接是0,有优化空间

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI0NTE2NTQ5_size_16_color_FFFFFF_t_70]: /images/20220405/f10e0d0b272a42bcb1395d431ed1b219.png