【LeetCode】Spiral Matrix II

最新推荐文章于 2019-05-14 09:57:51 发布

原创最新推荐文章于 2019-05-14 09:57:51 发布 · 322 阅读

0 ·

CC 4.0 BY-SA版权

算法学习同时被 2 个专栏收录

104 篇文章

订阅专栏

【LeetCode】

100 篇文章

订阅专栏

本文介绍了一种使用C++编程语言实现生成指定大小的螺旋矩阵的方法，矩阵中的元素从1填充到n^2，遵循螺旋顺序排列。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

Given an integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.

For example,
Given n = 3,

You should return the following matrix:

[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

跟 Spiral Matrix差不多，用个变量记录下写到哪了就好了。

代码：

class Solution {
public:
	vector<vector<int> > generateMatrix(int n) {
		vector<vector<int>> ret(n, vector<int>(n));
		int N = n*n;
		int rowstart(0), colstart(-1), rowend(n - 1), colend(n - 1);
		int curr(1);
		while (curr <= N){
			if (++colstart > colend) break;
			for (int i = colstart; i <= colend; i++){
				ret[rowstart][i] = curr;
				curr++;
			}
			if (++rowstart > rowend) break;
			for (int i = rowstart; i <= rowend; i++){
				ret[i][colend] = curr;
				curr++;
			}
			if (--colend < colstart) break;
			for (int i = colend; i >= colstart; i--){
				ret[rowend][i] = curr;
				curr++;
			}
			if (--rowend < rowstart) break;
			for (int i = rowend; i >= rowstart; i--){
				ret[i][colstart] = curr;
				curr++;
			}
		}
		return ret;
	}
};