牛客网刷题之蛇皮走位的矩阵（BC134 蛇形矩阵）

题目描述

给你一个整数n，输出n∗n的蛇形矩阵。

输入描述：

输入一行，包含一个整数n

输出描述：

输出n行，每行包含n个正整数，通过空格分隔。

1<=n<=1000

举例

输入：4

输出：

1  2  6  7
3  5  8  13
4  9  12 14
10 11 15 16

分析：

通过对蛇皮走位的分析，发现蛇皮的前进方向一共分为四种，分别是向右，向下，右上，左下。

运动规则分为两种

碰撞上下两边时，向右前进一步，然后离开边界。

碰撞左右两边时，向下前进一步，然后离开边界。

其中碰撞的后在边界前进一步后会以反方向离开。

没有碰撞时，则是按照右上或者坐下的方向在方块内进行运动。

所以蛇皮的移动可以分为两部分。

第一部分：可以设置一个方向变量flag，假设flag为1时，蛇皮向右上方行动，flag为0时，蛇皮向左下方行动。

第二部分：蛇皮碰到边界时的运动

int main()
{
	int num, row = 0, col = 0, flag = 1, count = 2, arr[50][50];
	scanf("%d", &num);
	arr[row][col] = 1;
	while (count <= num*num)
	{
		if (flag == 1 && row == 0 && col < num - 1)//碰到上边界，向右一步，然后调整方向为左下
		{
			arr[row][++col] = count;
			flag = 0;
		}
		else if (flag == 0 && row == num - 1)//碰到下边界，向右一步，然后调整方向为右上
		{
			arr[row][++col] = count;
			flag = 1;
		}
		else if (flag == 0 && col == 0)//碰到左边界，向下一步，然后调整为右上
		{
			arr[++row][col] = count;
			flag = 1;
		}
		else if (flag == 1 && col == num - 1 && row < num - 1)//碰到右边界，向下一步，然后调整为左下
		{
			arr[++row][col] = count;
			flag = 0;
		}
		else if (flag == 1)//向右上移动
		{
			arr[--row][++col] = count;
		}
		else if (flag == 0)//向左下移动
		{
			arr[++row][--col] = count;
		}
		count++;
	}
	for (int i = 0; i < num; i++)
	{
		for (int j = 0; j < num; j++)
		{
			printf("%d ", arr[i][j]);
		}
		printf("\n");
	}
	system("pause");
	return 0;
}