杨辉三角求某个数 n 在第几个数出现? n < = 1 0 9 n<=10^9 n<=109
(1)
观察发现, 1 0 9 10^9 109 比较坏的情况下出现在什么位置: C 45000 2 > 1 0 9 C^2_{45000} > 10^9 C450002>109 如果这个位置都找不到,那么肯定在
C n 1 C^1_{n} Cn1 处 , 而且 C 32 16 < 1 0 9 但 C 32 17 > 1 0 9 C^{16}_{32} < 10^9 但 C^{17}_{32} > 10^9 C3216<109但C3217>109
因此,需要存一下杨辉三角
d
p
[
45000
]
[
17
]
dp[45000][17]
dp[45000][17] ,其中
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j] 表示第i行,第j列是多少
如果这个数大于
1
0
9
10^9
109 不用更新,因为下边不会再用到这个数
#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;
}