蓝桥杯 杨辉三角形 C/C++ B组
感觉这道题很坑,它范围给的很大,先开始做一直超时,后面也是看了另一个博客才搞出来,下面附上链接
https://blog.csdn.net/Huo6666/article/details/123284403
#include <iostream>
using namespace std;
/*
思路:定义两个数组,a,b.a用来存储三角形的第n行,b存储第n-1行(我的方法是同时计算和更新b).
因为杨辉三角形是对称的,所以每一行其实只需要计算一半多一位(速度会快很多),
多一位是因为第n行中间位是第(n+1)/2位,它的下一行可能是第(n+1)/2+1位,
需要依赖第n行的第(n+1)/2+1位和中间位相加计算得到(理解不了可以用笔计算几行) ,
为了避免出问题都使用了long long
*/
int main(){
long long N;
cin >> N;
//44900 原因见上面另一个博客链接
long long b[44900];
long long a[44900];
//初始化b为杨辉三角的第二行,a的第一位为1
b[1] = b[0]=a[0]=1;
bool condition = true;
//i代表a的行数,index代表最终输出位置
long long i = 3;
long long index;
//N为1直接输出
if(N==1)
{
cout << 1 <<endl;
return 0;
}
//第一次循环是依据b计算a,a此时代表的是杨辉的第3行
while (condition){
//j最大是中间位(终止条件)
for (long long j =1;j<=(i+1)/2-1;j++){
a[j] = b[j]+b[j-1];
//因为b[j-1]已经失去作用了,所以可以直接更新
b[j-1]=a[j-1];
//计算到中间位(即将终止循环时)的时候要更新b[j],和b[j+1]
if(j==(i+1)/2-1){
b[j]=a[j];
//根据杨辉三角对称性得到
b[j+1]=a[i-(j+2)];
}
//符合条件就输出
if (a[j]==N){
index=(i-1)*i/2+j+1;
cout << index <<endl;
return 0;
}
}
i++;
//44800终止循环,原因见上面另一个博客链接
if(i>44800)
condition=false;
}
index=(N+1)*N/2+2;
cout<<index<<endl;
return 0;
}