第十二届蓝桥杯B组C++杨辉三角

传送门 嗖啪 杨辉三角
题意
给你一个n，问你这个n出现的位置

首先这个数的规模特别大，暴力肯定会T
那么你就要开始找规律，有什么规律呢
你分成两半看

C(N, M)在这里是求组合数，n在下，m在上
这样斜着看，能看出来最中间那一竖行最小，把他们看作那一斜行的开始，然后再找到规律，中间一竖行分别是C（0，0）， C（2，1）， C（4，2）；也就是C2k k， 他们斜着那一行也是C（n， k）, 这n要大于2k，根据计算，n最大也就是C（32， 16），也就是你只需要枚举十六行，你就能找到答案
我们可以发现这个组合数，第零号是有一个数，第一行是有两个数
假设答案是C（r， k），也就是r前边有r行，然后前r行的所有个数就是r*(r+1)/2;第r+1行从0开始数，然后一直到k就是k+1

上代码

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <map>
#include <string>
#include <cstring>
#include <cmath>
#include <stack>
using namespace std;
#define INF 0x3f3f3f3f
#define ll long long
#define fast ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define sc(a) scanf("%lld",&a)
#define pf(a) printf("%d",a) 
#define endl "\n"
#define int long long
int n;
int c(int a, int b)//算排列组合 
{
	int res=1;
	for (int i = a, j = 1; j <= b; i --, j ++ )
    {
        res = res * i / j;
        if (res > n) return res;
    }
    return res;
	
} 


bool check(int k)
{
	int l = k*2, r=max(n, l);//l是左端点，因为这里最小也要是2k，所以让l=2k，而n肯定在C（n, k）到C(2K, K)中
	while(l <= r)
	{
		int mid = l+r >> 1;
		if(c(mid, k) == n) 
		{
			if(n==1) cout<<1<<endl;
			else cout<< r*(r+1) /2 + k + 1;//根据手动模拟，这里n==1不符合这个公式
			return 1;
		}
		else if(c(mid, k) > n) r = mid-1;
		else l = mid+1;
	}
	if(c(r,k) != n) return 0;
	
//	cout<< r*(r+1) /2 + k + 1;
//	
//	return 1;  
}

/*
//这里也可以用这个代码，就不会出现单独判断的情况了
bool check(int k)
{
    LL l = k * 2, r = max((LL)n, l);
    while (l < r)
    {
        LL mid = l + r >> 1;
        if (C(mid, k) >= n) r = mid;
        else l = mid + 1;
    }
    if (C(r, k) != n) return false;

    cout << r * (r + 1) / 2 + k + 1 << endl;

    return true;
}
*/
signed main()
{
//	int a, b;
//	while(cin>>a>>b)
//	cout<<c(a, b);
	sc(n);
	for(int k=16;;k--)
	{
		if(check(k)) break;
	}
	return 0;
}