《代码随想录》学习笔记---螺旋矩阵Ⅱ

一、题目

给你一个正整数 n ,生成一个包含 1 到 n^{2} 所有元素,且元素按顺时针顺序螺旋排列的 n*n 正方形矩阵 matrix 。

【示例】输入n=3时,得到如下矩阵

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

二、思路及要点

主要依赖四个方向的循环实现题目要求(两行两列),需要注意临界时的条件判断。

通过四个变量控制:

start_line, end_line, start_col, end_col

如果n为奇数,则矩阵正中央的数循环中是填不进去的,需要特殊判断 

if (n % 2 == 1)
{
	a[(n - 1) / 2][(n - 1) / 2] = cnt;
}

三、代码

class Solution {
public:
	vector<vector<int>> generateMatrix(int n) {
		vector<vector<int>> a(n,vector<int>(n,0));
		int start_line = 0, end_line = n - 1;
		int start_col = 0, end_col = n - 1;
		int cnt = 1;
		while (cnt < (n * n))
		{
			for (int j = start_col; j <= end_col; j++)
			{
				a[start_line][j] = cnt;
				cnt++;
			}
			start_line++;
			for (int i = start_line; i <= end_line; i++)
			{
				a[i][end_col] = cnt;
				cnt++;
			}
			end_col--;
			for (int j = end_col; j >= start_col; j--)
			{
				a[end_line][j] = cnt;
				cnt++;
			}
			end_line--;
			for (int i = end_line; i >= start_line; i--)
			{
				a[i][start_col] = cnt;
				cnt++;
			}
			start_col++;
		}
		if (n % 2 == 1)
		{
			a[(n - 1) / 2][(n - 1) / 2] = cnt;
		}
		return a;
	}
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值