每日一题洛谷P8749 [蓝桥杯 2021 省 B] 杨辉三角形c++

#include <iostream>
#define int long long
using namespace std;
int n;
// 计算组合数C(a, b)
int C(int a, int b) {
    int res = 1;
    for (int i = a, j = 1; j <= b; i--, j++) { // 循环计算组合数
        res = res * i / j; // 依次相乘并除以j
        if (res > n) return res; // 若超过n则提前返回，防止溢出
    }
    return res;
}
signed main() {
    cin >> n;
    if (n == 1) {
        cout << 1 << endl;
        return 0;
    }
    //1e9的数据在第16斜行之内
    for (int i = 16; i >= 0; i--) {
        int l = 2 * i; // 斜行i的最小行号为2i
        int r = 1e9; // 设置一个足够大的右边界
        //二分查找
        while (l <= r) {
            int mid = l + (r - l) / 2;
            int lim = C(mid, i); // 计算组合数C(mid, i)
            if (lim == n) { // 找到目标组合数
                // (mid+1)*mid/2为前mid行的元素总数，i+1为当前行的位置
                cout << (mid * (mid + 1)) / 2 + i + 1 << endl;
                return 0;
            }
            else if (lim < n) l = mid + 1;
            else r = mid - 1;
        }
    }
    return 0;
}