2020第11届蓝桥杯C++B组蓝桥2020试题 第二套C: 蛇形填数

第三题：蛇形填数

题目描述
如下图所示，小明用从1开始的正整数“蛇形”填充无限大的矩阵。容易看出矩阵第二行第二列中的数是5。请你计算矩阵中第20行第20列的数是多少？

参考了大佬的写法，原来这个蛇形填数在算法竞赛书上也提到过，但是由于刚接触，当时考的时候没看出来，硬画图数的，好像最后还数对了

#include<iostream>
using namespace std;

int main(){
int i=0;
int j=0;
int cnt=2;//用来为每个元素赋值
//虽然说是每个元素都跟上一个元素多1，但是坐标不好定位 
int a[250][250];
a[0][0]=1;

while(cnt<1000){
	//这个是按照斜线数字增长的顺序来的
	//第一个while是斜向下的顺序
	//第二个while是斜向上的顺序 
	//切换顺序为某一横/纵坐标为0 
j++;
while(i!=-1&&j!=-1){
	a[i][j]=cnt++;
	if(j==0)
	break;
	i++;
	j--;
}
i++;
while(i!=-1&&j!=-1){
	a[i][j]=cnt++;
	if(i==0)
	break;
	i--;
	j++; 
} 
}

cout<<a[19][19];
	return 0;
} 

根据评论区大佬的提示

可以看出要求的元素正好在对角线上
而对角线的元素正好符合下一个比上一个多i-1个4
所以代码就可以简化成

#include<iostream>
using namespace std;
int sum;
int main(){
	int n=1;
	for(int i=2;i<=20;i++){
		n+=(i-1)*4;
		
	}
	cout<<n<<endl;
	return 0;
} 

第一次蓝桥杯的时候真的什么也不知道，硬是做题，
整的老难受了
实际上啥也不会