AcWing 3418. 杨辉三角形

杨辉三角求某个数 n 在第几个数出现？ $n<=10^9$

（1）

观察发现，$10^9$ 比较坏的情况下出现在什么位置：$C_{45000}^2>10^9$ 如果这个位置都找不到，那么肯定在

$C_n^1$ 处 ， 而且 $C_{32}^{16}<10^9但C_{32}^{17}>10^9$

因此，需要存一下杨辉三角
$dp[45000][17]$ ，其中$dp[i][j]$ 表示第i行，第j列是多少
如果这个数大于$10^9$ 不用更新，因为下边不会再用到这个数

#include <iostream>
using namespace std;

#define LL long long
LL dp[45000][101];

int main()
{
	LL n;
	cin >> n;
	if(n==1) {
		cout << 1;
		return 0;
	}
	dp[1][1] = 1;
	for(int i = 2; i <= 45000; i++) {
		for(int j = 1; j <= min(i, 17); j++) {
			if(dp[i-1][j-1] + dp[i-1][j] <= 1e9)
				dp[i][j] = dp[i-1][j-1] + dp[i-1][j];
			if(dp[i][j] == n) {
				cout << (i-1) * i / 2 + j; 
				return 0;
			}
		}
	}

	cout << (n+1)*n/2 + 2;
	return 0;
}

(2)

二分

#include <bits/stdc++.h>
using namespace std;

#define LL long long

LL n;

LL cal(LL p, LL k)
{	
	LL res = 1;
	for(LL i = 1; i <= k; i++) 
	{
		res = res * p / i;
		p--;
		if(res > n) return res; 
	}
	return res;
}

bool check(LL i)
{
	LL l = 2 * i, r = max(n, l);
	
	while(l < r) {
		LL mid = (l + r) >> 1;
		if(cal(mid,i) >= n) r = mid;
		else l = mid + 1;
	}
	if(cal(r,i)==n) {
		cout << (r+1)*r/2 + i + 1;
		return true;
	}
	return false;
}

int main()
{
	cin >> n;
	
	for(LL i = 16; i >= 0; i--)
		if(check(i)) break;

	return 0;
}