求 N* M阶螺旋矩阵

问题描述

按顺时针方向构建一个MxN的螺旋矩阵（或按顺时针方向螺旋访问一个N* M的矩阵）例如：N=4， M=5时
1 2 3 4 5
14 15 16 17 6
13 20 19 18 7
12 11 10 9 8

思路

从[1, 0]开始，按照→，↓，←，↑4个方向， 判断该点的值是否为0， 若为0，依则填入count++（初值为1）,否则改变方向，代码如下：
Python代码：

matrix = [[0 for i in range(101)] for i in range(101)]   # 用 0 初始化一个足够大的二维矩阵
N, M = 4, 5
matrix[1][M + 1], matrix[N + 1][M], matrix[N][0] = -1, -1, -1   # 设置自三个点作为边界，也可设置一圈，用于最外圈改变方向  
count = 1   # 计数
dr, dc = 0, 1     # dr = [0, 1, 0, -1] 控制方向
                 # dc = [1, 0, -1, 0]
i, j = 1, 0  # 矩阵索引  从matrix[1][0]开始
num = N * M   # 算出所求螺旋矩阵的元素个数（代码外提）
while count <= num:
    if matrix[i + dr][j + dc] == 0:
        matrix[i + dr][j + dc] = count  # 赋值
        count += 1
        i += dr   # 进入下一个点
        j += dc
    else:
        dr, dc = dc, -dr  # 依次改变方向 注意c++ 和python在此处的区别 

for i in range(1, N + 1):   # 输出螺旋矩阵
    for j in range(1, M + 1):
        print(matrix[i][j], end="\t")
    print()

c++

#include<iostream>
using namespace std;

int matrix[101][101] = {{0}};
const int N = 3;
const int M = 4;
const int num = N * M;

int main()
{
	matrix[1][M + 1] = -1;
	matrix[N + 1][M] = -1;
	matrix[N][0] = -1;
	int count = 1;
	int i = 1, j = 0;
	int dr = 0, dc  = 1;
	
	while(count <= num)
	{
		if(matrix[i + dr][j + dc] == 0)
		{
			matrix[i + dr][j + dc] = count++;
			i += dr;
			j += dc;
		}
		else
		{
			int t = dr;
			dr = dc;
			dc = -t;
		}
	}
	
	for(int i = 0; i<=N+1; i++)
	{
		for(int j = 0; j<=M+1; j++)
		{
			cout << matrix[i][j] << "\t"; 
		}
		cout << endl;
	}
	return 0;
} 

若有错误或者可以改进的地方，还请大家指出。