蓝桥杯杨辉三角

最新推荐文章于 2024-10-06 14:10:45 发布

NP_hard

最新推荐文章于 2024-10-06 14:10:45 发布

阅读量590

点赞数

分类专栏： Blue Bridge Cup

本文链接：https://blog.csdn.net/NP_hard/article/details/123670878

版权

组合计算二分查找 C++ 算法优化数学问题

关键词由CSDN通过智能技术生成

Blue Bridge Cup 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

文章目录

题目
代码

题目

参考链接
在这里插入图片描述

代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
int n;
LL C(int a, int b)   //计算C（a，b）
{
    
    LL res = 1;
    for(int i = a, j = 1; j <= b; i --, j ++)
    {
    
        res = res * i / j;
        if(res > n)
            return res;     // 大于n已无意义，且防止爆LL
    }
    return res;
}
bool check(int k)
{
    
    // 二分该斜行,找到大于等于该值的第一个数
    // 左边界2k，右边界为max(l, n)取二者最大，避免右边界小于左边界
    int l = 2 * k, r = max(n,l);
    while(l < r)
    {
    
        int mid = l + r >> 1;
        if(C(mid, k) >= n) r = mid;
        else l = mid + 1;
    }
    if(C(r, k) != n)
        return false;
    cout << 1ll*(r + 1) * r / 2 + k + 1 << endl;
    return true;
}
int main()
{
    
    cin >> n;
    // 从第16斜行枚举
    for(int i = 16; ; i--)
        if(check(i))
            break;
    return 0;
}