百炼：C:蛇形填充数组

C:蛇形填充数组

• 查看
• 提交
• 统计
• 提问

总时间限制: 

1000ms 

内存限制: 

65536kB

描述

用数字1,2,3,4,...,n*n这n²个数蛇形填充规模为n*n的方阵。

蛇形填充方法为：

对于每一条左下-右上的斜线，从左上到右下依次编号1,2,...,2n-1；按编号从小到大的顺序，将数字从小到大填入各条斜线，其中编号为奇数的从左下向右上填写，编号为偶数的从右上到左下填写。

比如n=4时，方阵填充为如下形式：

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

输入

输入一个不大于10的正整数n，表示方阵的行数。

输出

输出该方阵，相邻两个元素之间用单个空格间隔。

样例输入

4

样例输出

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

#include<iostream>
#include<vector>
#include<string>

using namespace std;
int main()
{

	
	int n;
	cin >> n;
	if (n == 1)
	{
		cout << 1 << endl;
		return 0;
	}
	vector<vector<int>> m(n,vector<int>(n, 0));
	int i = 0,j = 0;
	int flag = 1;  //flag为1表示方向向右上，-1表示方向向左下
	m[0][0] = 1;

	for (int c = 2; c <= n*n; c++)  //依次填充2~n*n
	{
		if (flag == 1)
		{
			if (i - 1 >= 0 && j + 1 < n)//能否向右上走
			{
				i--;
				j++;
				m[i][j] = c;
			}
			else if (j + 1 < n)//不能右上的话能否向右走
			{
				j++;
				m[i][j] = c;
				flag = -flag;
			} 
			else if (i + 1 < n)//不能向右的话能否向下走
			{
				i++;
				m[i][j] = c;
				flag = -flag;
			}
		}
		else
		{
			if (i + 1 < n&&j - 1 >= 0)  //能否向左下走
			{
				i++;
				j--;
				m[i][j] = c;

			}
			else if (i + 1 < n) //左下不行的话能否直接向下走
			{
				i++;
				m[i][j] = c;
				flag = -flag;
			}
			else if (j + 1 < n)//向下不行的话能否向右走
			{
				j++;
				m[i][j] = c;
				flag = -flag;
			}
		}
	}

	for (int i = 0; i < n; i++)//输出结果
	{
		for (int j = 0; j < n; j++)
		{
			if (j != n - 1)
				cout << m[i][j] << " ";
			else
				cout << m[i][j] << endl;
		}
	
	}
	system("pause");
}