第十一届蓝桥杯第二场省赛：蛇形矩阵

如下图所示，小明用从 1 开始的正整数“蛇形”填充无限大的矩阵。
1 2 6 7 15 :::
3 5 8 14 :::
4 9 13 :::
10 12 :::
11 :::
:::
容易看出矩阵第二行第二列中的数是 5。请你计算矩阵中第 20 行第 20 列
的数是多少 ?

模拟题：在考试的时候记得曾经写过，但是忘记具体怎么写了，后悔。

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

 
int main(int argc, char** argv) {
	int ans[50][50];
	int n = 45;
	int num = n*n;
	for(int i = 1, cnt = 1; i <= n && cnt <= num; i++)
	{
		if(i&1)
		{
			for(int x = i, y = 1; x >= 1 && y <= i; x--,y++)
				ans[x][y] = cnt++;
		}else{
			for(int x = 1, y = i; x <= i && y >= 1; x++,y--)
				ans[x][y] = cnt++;
		}	
	 } 
	
	for(int i = 1; i <= 20; i++)
	{
		for(int j = 1; j <= 20; j++)
			cout << ans[i][j] << " " ;
		cout << endl;
	}
	
	cout << ans[20][20] << endl;
	
	return 0;
}