P4370-[Code+#4]组合数问题2【数学,堆】

1|0正题

题目链接:https://www.luogu.com.cn/problem/P4370

---

1|1题目大意

求满足$m\leq n\leq a$的情况下，前$k$大的$\binom{n}{m}$的和。

$1\leq n\leq 10^6,1\leq k\leq 10^5$

---

1|2解题思路

首先想到的是$\binom{n}{m}>\binom{n-1}{m}$（这个十分显然）。

第一想法是开$n$个堆然后每次取最大的，但是发现我们不能比较组合数的大小。

然后看题解发现可以直接取组合数的$log$这样就能比较了。

时间复杂度：$O((k+n)\log n)$

---

1|3code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define ll long long
using namespace std;
const ll N=1e6+10,P=1e9+7;
struct node{
	double w;ll x,y;
};
bool operator<(node x,node y)
{return x.w<y.w;}
ll n,k,ans,inv[N],fac[N];
double s[N];
priority_queue<node> q;
double D(ll n,ll m)
{return s[n]-s[m]-s[n-m];}
ll C(ll n,ll m)
{return fac[n]*inv[m]%P*inv[n-m]%P;}
signed main()
{
	inv[0]=inv[1]=fac[0]=1;
	for(ll i=2;i<N;i++)inv[i]=P-inv[P%i]*(P/i)%P;
	for(ll i=1;i<N;i++){
		inv[i]=inv[i-1]*inv[i]%P;
		fac[i]=fac[i-1]*i%P;
		s[i]=s[i-1]+log(i);
	}
	scanf("%lld%lld",&n,&k);
	for(ll i=0;i<=n;i++)
		q.push((node){D(n,i),n,i});
	while(k--){
		node x=q.top();q.pop();
		(ans+=C(x.x,x.y))%=P;
		if(x.x==x.y)continue;
		q.push((node){D(x.x-1,x.y),x.x-1,x.y});
	}
	printf("%lld\n",ans);
	return 0;
}

__EOF__

本文作者：QuantAsk
本文链接：https://www.cnblogs.com/QuantAsk/p/15410014.html
关于博主：退役OIer，GD划水选手
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！