面試題——蛇形矩陣

AcWing 756 蛇形矩陣

題目

輸入兩個整數n和m，輸出一個n行m列的矩陣，將數字1到n * m按照回字蛇形填充至矩陣中。

具體矩陣形式可參考樣例。

輸入格式

輸入共一行，包含兩個整數n和m。

輸出格式

輸出滿足要求的矩陣。

矩陣佔n行，每行包含m個空格隔開的整數。

數據範圍

1≤n,m≤1001≤n,m≤100

輸入樣例：

3 3

輸出樣例：

1 2 3
8 9 4
7 6 5

思路

​ 這道題是要將大小爲1 ~ n * m的數字按照順時針繞圈（有點白話）的方式放入到一個n * m的數組中，所以做法十分明顯，我們需要定義_右下左上四個方向的偏移量_，通過在現有座標上加上方向偏移量來實現我們當前指針（也就是二維數組下標）的移動

​ 現在怎麼移動的問題解決了，另一個問題就是判斷我們_下一次要移動的位置是否已經超出矩陣邊界_，或者_下一次要移動的位置是否已經被賦過值_了，通過這兩個條件，我們可以判斷我們的座標移動是不是需要_拐彎_了，如果是，只需要讓偏移量改變即可，另外由於矩陣中數字的範圍是1 ~ n * m，所以凡是下一個位置檢測到!= 0的數字，那就說明這個位置已經被走過了

​ 下面我們一起來看代碼

代碼

#include<iostream>

using namespace std;

int res[110][110];

int main() {
	int n, m;
	int i = 0, j = -1;
	int item = 0;
	int dx[4] = { 0, 1, 0, -1 };
	int dy[4] = { 1, 0, -1, 0 };
	scanf("%d%d", &n, &m);
	int x = 1;
	while (x <= n * m) {
		if (i + dx[item] >= n || j + dy[item] >= m || j + dy[item] == -1 || res[i + dx[item]][j + dy[item]]) {
			item++;
		}
		else {
			i += dx[item];
			j += dy[item];
			res[i][j] = x;
			x++;
		}
		if (item == 4)
			item = 0;
	}
	for (i = 0; i < n; i++) {
		for (j = 0; j < m; j++)
			printf("%d ", res[i][j]);
		puts("");
	}
}