5.力扣题目:59. 螺旋矩阵 II
解析:(循环不变量原则)
题目要求:给定一个正整数n,生成一个包含1到n平方的按照顺时针螺旋排列的矩阵。
该题不涉及任何特有的算法,就是按照题目的要求去模拟螺旋矩阵。
需要转多少圈?从矩阵中心点到边界的距离就是圈数,也就是边长的一半:n/2,对于n偶数情况,就是边长的一半。n为奇数,只是n/2圈数后,剩下的就是中心的一个空格是空的,这个好处理,直接补上最后一个空格。所以:圈数loop是n/2。走完圈数后,n为偶数则完成填充,n为奇数则补上中心还空的一个格子。
所以在转圈时,边界条件是loop<n/2,圈数loop从0开始。(为什么不从1开始,后面写代码时会发现,loop从0开始,loop=转圈时列数的起始位置)
每转一圈,++loop,同样转圈时列数的起始位置也要++,所以可以直接用loop指代转圈时列数的起始位置。
由于转圈时,存在多个边界问题,一定要遵循:循环不变量原则,始终坚持头闭尾开,就不会出错。
矩阵上侧从左到右,在loop行,每次从loop列开始填充(包括头部loop列),到n-loop-1列(不包括尾部n-loop-1列)填充元素。
矩阵右侧从上到下,在n-loop-1列,每次从loop行开始填充(包括头部loop行),到n-loop-1行(不包括尾部n-loop-1行)填充元素。
矩阵下侧从右到左,在n -loop-1行,每次从n-loop-1列(包括n-loop-1列),到loop列(不包括loop列)填充元素。
矩阵左侧从下到上,在loop列,每次从n-loop-1行(包括n-loop-1行),到loop行(不包括loop行)填充元素。
最后走完圈数后,n为偶数则完成填充,n为奇数则补上中心还空的一个格子。
注意点:
这类边界问题易出错的题目,比如先前二分查找这类问题,一定要严格遵循:循环不变量原则 ,对于边界处理实行左闭右开或者左闭右闭等不变化的统一处理政策。不可以对处理边界的策略变来变去,分情况来改变边界处理策略容易出错。
代码:
class Solution {
public int[][] generateMatrix(int n) {
int loop = 0, i, j, count = 0;
int[][] res = new int[n][n];
while (loop < n/2) {
int len = n;
for (j = loop; j < n - loop - 1; j++) {
res[loop][j] = ++count;
}
for (i = loop; i < n - loop -1; i++) {
res[i][n - loop -1] = ++count;
}
for (; j > loop; j--) {
res[n - loop -1][j] = ++count;
}
for (; i > loop; i--) {
res[i][loop] = ++count;
}
loop++;
}
if (n % 2 == 1) res[loop][loop] = ++count;
return res;
}
}