第十二届蓝桥杯真题-杨辉三角形(二分)

题目描述

下面的图形是著名的杨辉三角形：

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

输入

输入一个整数 N。

输出

输出一个整数代表答案。

样例输入

6

样例输出

13

思路

看到这个题的第一想法就是递推，建一个足够大的一维数组，然后一层一层的往下推，直到遇到了那个数停下来，再计算位置输出，写出来后勇敢交一发，果不其然，TLE了！ T_T

再仔细想一想，既然连遍历都过不了，那应该是有什么快速方法可以迅速定位，顿时想到了二分，但是二分要求数字递增排列，这看着也不太像(除非斜着看)，后面经过大佬提醒，发现斜着看竟然是组合数，如果从0开始标，那么第i行第j个的值就是Cij(以i为底的组合数)。这么看就好写了。我们可以一列一列进行二分查找，从后往前可以更快的找到，但是注意从后往前找的话只能找左半边的三角，比如样例图中的10，从后往前找后面的10会先找到，所以我们找第j列的时候就从第2*j行开始找，这样保证了只找左半边。

代码

#include "iostream"
#include "cstdio"
using namespace std;
const int N=1000010;
long long n;
long long ask(int q,int p)//求组合数
{
	long long res=1;
	for(int i=1,j=q-p+1;i<=p;i++,j++)
	{
		res=res*j/i;
		if(res>n) return res;
	}
	return res;
}
int check(int x)
{
	long long l=2*x,r=n,mid;//只找左半边
	while(l<=r)
	{
		mid=(l+r)>>1;
		long long num=ask(mid,x);
		if(num>n) r=mid-1;
		else if(num<n) l=mid+1;
		else
		{
			long long res=mid*(mid+1)/2+x+1;
			cout<<res<<endl;
			return 1;
		}
	}
	return 0;
}
int main()
{
	cin>>n;
	for(int i=14;i>=0;i--) if(check(i)) break;
}