杨辉三角形(第十二届蓝桥杯 省赛 B组）

想要理解的去看
这个博主的详解

【问题描述】

下面的图形是著名的杨辉三角形：
如果我们按从上到下、从左到右的顺序把所有数排成一列，可以得到如下
数列：
1, 1, 1, 1, 2, 1, 1, 3, 3, 1, 1, 4, 6, 4, 1, …
给定一个正整数 N，请你输出数列中第一次出现 N 是在第几个数？

【输入格式】

输入一个整数 N。

【输出格式】

输出一个整数代表答案。

【样例输入】

6

【样例输出】

13

【评测用例规模与约定】

对于 20% 的评测用例，1 ≤ N ≤ 10；
对于所有评测用例，1 ≤ N ≤ 1000000000。

代码

#include "stdio.h"
#include "algorithm"
#include "iostream"
using namespace std;
typedef long long ll;
ll n;
ll C(ll a,ll b)
{
	ll sum=1;
	for(int i=a,j=1;j<=b;i--,j++)
	{
		sum=sum*i/j;//i的每个值都比j大，因为2*b<=a
		if(sum>n)//大于n,无意义,继续算下去可能爆long long
			return sum;
	}
	return sum; 
}
ll find(ll k)
{
	ll l=2*k,r=max(l,n);//l左边界，r右边界
	//C(2*k,k)代表第k个斜行最小的值(第一个值)
	//C(2*k+1,k)代表第k个斜行第二小的值(第二个值)以此类推
	while(l<=r)
	{
		
		ll mid=(r+l)/2;
		ll s=C(mid,k);
		if(s>n) r=mid-1;
		else if(s<n) l=mid+1;
		else  return 1ll*mid*(mid+1)/2+k+1;
	}  
	return 0;
}
int main()
{
	
	cin>>n;
	for(int i=16;i>=0;i--)//代表第i个斜行
	{
		ll sum=find(i);
		if(sum)
		{
			cout<<sum;
			return 0;
		}		
	}
}