【贪心】【大数比较大小】组合数问题

【描述】
给你两个数 n 和k ，希望你找到 k 个不同的组合数使得这 k个组合数的和最大。所谓不同的组 合数，即对于组合数 和 ，若a1!=a2 或者 b1!=b2，则我们认为这两个组合数是不同 的。现在小葱希望你找到这样 k 个不同的组合数，使得它们互不相同且对于其中任何一个组 合数 有0<=a<=b<=n 。问这 k个组合数的和最大是多少？

【输入】
第一行两个整数 n,k。

【输出】
一行一个整数，代表 k 个组合数的和对 1000000007取模之后的结果；数据保证一定有至少 k个数可以选。

【样例输入】
2 3
【样例输出】
4

【思路】

首先，我们可以通过打表发现组合数的大小大致是一个山峰状，满足某种单调性。所以我们可以从${(}_n^{2n})$开始，维护一个堆，每次取出最大的，然后把它周围几个组合数丢进堆里面。问题只剩下如何比较大小。这是一个套路，我们可以取对数进行比较大小，把组合数的阶乘变为若干个数相加，这样就可以实现比较大小了。如果题目中涉及判断相等，可以再维护一个hash进行判断。当然，如果担心组合数可能不满足这样的山峰形状，有一个更严谨的做法。杨辉三角形中由下向上是递增的，所以我们维护每一列中最大的数，从堆里取出后放入该列下一个数。
代码：

#include<bits/stdc++.h>
#include<tr1/unordered_map>
#define re register
using namespace std;
const int N=2e6+5;
const int mod=1e9+7;
int n,k,a,b;
tr1::unordered_map<long long,bool>f;
inline int red(){
    int data=0;int w=1; char ch=getchar();
    while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0' && ch<='9') data=(data<<3)+(data<<1)+ch-'0',ch=getchar();
    return w==1?data:-data;
}
long double fac[N];
inline int mul(const int&a,const int&b){return 1ll*a*b%mod;}
int fc[N]={1},ifc[N]={1,1},inv[N]={0,1};
struct node{
	long double w;int a,b;
	friend inline bool operator<(const node&a,const node&b){return a.w<b.w;}
};
priority_queue<node>q;
int ans=0;
void pre(){
	for(int re i=1;i<=n;i++)
		fac[i]=fac[i-1]+log(i),fc[i]=mul(fc[i-1],i);
	for(int re i=2;i<=n;i++){
		inv[i]=mul(inv[mod%i],mod-mod/i);
		ifc[i]=mul(ifc[i-1],inv[i]);
	}
}
inline long long hash(const int&a,const int&b){return 1000000ll*a+b;}
inline int C(const int&a,const int&b){return mul(fc[a],mul(ifc[a-b],ifc[b]));}
inline void push(const int&a,const int&b){
	if(a<b||a>n||b<0||f[hash(a,b)])return;f[hash(a,b)]=1;
	q.push((node){fac[a]-fac[b]-fac[a-b],a,b});
}
int main(){
	n=red();k=red();pre();
	if(n&1)push(n,n>>1),push(n,(n>>1)+1);
	else push(n,n>>1);
	while(k--){
		node t=q.top();q.pop();
		if((ans+=C(t.a,t.b))>=mod)ans-=mod;
		push(t.a,t.b+1);push(t.a,t.b-1);
		push(t.a-1,t.b);push(t.a-1,t.b-1);
	}cout<<ans<<"\n";
}