晚上无聊翻看算法竞赛入门经典,发现了一道熟悉的题目-蛇形填数,当时大一上学期学得就一团乱麻,今晚研究了一下,终于写出来了还是很高兴的。
代码的核心在于四个while循环的约束条件,能保证蛇形在目标位置变换方向。先定义一个二维数组,然后将二维数组的左上角赋初始值为1,之后向下走,向左走,向上走,向上走。核心部分分别为: x + 1 < n; y - 1 >= 0; x - 1 >= 0; y + 1 < n.
具体代码如下:
#include<iostream>
#include<stdio.h>
#include<string>
#define max 10
using namespace std;
int a[max][max];
int main() {
int n, x, y, num;
memset(a, 0, sizeof(a));
scanf("%d", &n);
x = 0;
y = n - 1;
num = a[x][y] = 1;
while (num < n * n) {
while (x + 1 < n && a[x + 1][y] == 0) {
a[++x][y] = ++num;
}
while (y - 1 >= 0 && a[x][y - 1] == 0) {
a[x][--y] = ++num;
}
while (x - 1 >= 0 && a[x - 1][y] == 0) {
a[--x][y] = ++num;
}
while (y + 1 < n && a[x][y + 1] == 0) {
a[x][++y] = ++num;
}
}
for (x = 0 ; x < n ; x++) {
for (y = 0 ; y < n ; y++) {
printf("%3d", a[x][y]);
}
printf("\n");
}
return 0;
}