蓝桥杯 杨辉三角形 C/C++ B组

蓝桥杯 杨辉三角形 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;
}