洛谷 P4370 [Code+#4]组合数问题2 (逆元求组合数+优先队列)

题目描述

众所周知，小葱同学擅长计算，尤其擅长计算组合数，所以小葱给了你两个数 n和 k ，希望你找到 k 个不同的组合数使得这 k个组合数的和最大。所谓不同的组合数，即对于组合数 $C_{a_1}^{b_1}$和 $C_{a_2}^{b_2}$ ，若 $a_1\ne a_2$或者 $b_1\ne b_2$，则我们认为这两个组合数是不同的。现在小葱希望你找到这样k 个不同的组合数，使得它们互不相同且对于其中任何一个组合数 $C_a^b$
​有 $0\le b\le a\le n$ 。问这 k 个组合数的和最大是多少？
输入输出格式

输入格式：
从标准输入读入数据。

第一行两个整数 n,k 。

输出格式：
输出到标准输出。

一行一个整数，代表 k 个组合数的和对 10^9+7
取模之后的结果；数据保证一定有至少 k 个数可以选。

输入输出样例

输入样例#1：
2 3
输出样例#1：
4

思路

我们先写一个杨辉三角的图来观察一下
$$\begin{array}{cccccc}1& & & & & \\ 1& 1& & & & \\ 1& 2& 1& & & \\ 1& 3& 3& 1& & \\ 1& 4& 6& 4& 1& \\ 1& 5& 10& 10& 5& 1\\ ⋮& & & & & \end{array}$$
我们发现其实在n这一行以及这一行最大值的上一行的相同列元素为最优，所以我们可以用一个优先队列来维护一下最大值，先把n这一行的所有元素入队，优先队列中如果是相同值的先入队的排在前面，也就是说能保证取到是最大值他的上一行的同一列元素也是那一行的最大值，也就是说我们每取一个最大值就要把他的行减减，然后再重新计算他的$C_n^m$，由于n的范围较大，可以先对阶乘和阶乘的逆元打表，两个打表都是线性的，然后再套用组合数的公式就可以求出组合数取模后的值了，但是优先队列不可以用这个值进行大小比较，因为取模过了，所以我们可以用一个骚操作，我们把阶乘转化为对数加法的形式，那么两个组合数就可以用浮点数的方式来比较大小了

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <math.h>
#include <queue>
using namespace std;
const int N=1e6+5;
const int MAXN=1e6;
const long long mod=1e9+7;
long long fac[N],inv[N];
double L[N];
long long quickmod(long long a,long long b)
{
    long long ans=1;
    while(b)
    {
        if(b%2==1)
            ans=ans*a%mod;
        a=a*a%mod;
        b=b/2;
    }
    return ans;
}
struct node
{
    int n,m;
    double v;
    long long Cnm;
    node(int _n,int _m)
    {
        n=_n,m=_m,v=L[n]-L[m]-L[n-m];
        Cnm=fac[n]%mod*inv[m]%mod*inv[n-m]%mod;
    }
    friend bool operator <(node a,node b)
    {
        return a.v<b.v;
    }
};
priority_queue<node> q;
int main()
{
    fac[0]=fac[1]=1;
    for(int i=2; i<=MAXN; i++)fac[i]=fac[i-1]*i%mod;
    inv[MAXN]=quickmod(fac[MAXN],mod-2);
    for(int i=MAXN-1; i>=0; i--)inv[i]=inv[i+1]*(i+1)%mod;
    for(int i=1; i<=MAXN; i++)
        L[i]=L[i-1]+log(i);
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=0; i<=n; i++)
    {
        q.push(node(n,i));
    }
    long long ans=0;
    for(int i=1; i<=k; i++)
    {
        node temp=q.top();
        q.pop();
        ans=(ans+temp.Cnm)%mod;
        temp.n--;
        q.push(node(temp.n,temp.m));
    }
    printf("%lld\n",ans);
    return 0;
}