atcoder Beginner Contest 156 Roaming(组合数学 插板法)

题目大意:

有n个数,每个都是1,我们每次可以选择1个数减1,然后让另外一个数加1。但是我们不能生成负数。问我们假如进行k次操作,问我们这n个数能产生多少种不同的组合。注意[1,0,2]和[2,0,1]当作不同的组合。

3<=n<=2e5. 2<=k<=1e9.

解题思路:

看到n的范围,我们容易想到O(n),k次操作后,最多有min(k,n-1)个0。我们从0的个数开始下手,假设最终有m个零,我们可以证明剩下的n-m个数只需满足两个约束:

                                                                                 a_1,a_2,...a_{n-m}>=1

                                                                                 \sum_{i=1}^{n-m} a_i

其中第一个约束是我们假设决定的。第二个约束是题意决定的。

接下来,我们可以使用插板法往n个数里面插n-m-1个板,插板的结果就代表剩下的n-m个数的可能。比如5个数中假设有2个0,那么我们在5个数中间插2个板,得到2,2,1三种划分。这种求解约束条件转化为插板问题可以说是很巧妙了。

所以我们最终有公式:

\sum_{m=0}^{min(k,n-1)} C_n^m* C_{n-1}^{n-m-1}

m枚举的就是有几个0.

#include <bits/stdc++.h>
#define int long long 
using namespace std;
const int MODN=1e9+7;
const int MAXN=2e5+10;
int fact[MAXN];
int quick_pow(int a,int b,int modn=MODN){
    a%=MODN;
    int ret=1;
    // int cheng=a;
    while(b){
        if(b&1)ret*=a;
        ret%=MODN;
        a*=a;
        a%=MODN;
        b>>=1;
    }
    return ret;
}
int C(int n,int r){
    return fact[n]*(quick_pow((fact[n-r]*fact[r]),MODN-2))%MODN;
}

int32_t main(){
    int n,k;cin>>n>>k;
    fact[0]=1;
    for(int i=1;i<MAXN;i++)fact[i]=(fact[i-1]*i)%MODN;
    int sum=0;
    for(int i=0;i<=min(k,n-1);i++){
        sum+=C(n,i)*C(n-1,n-1-i);
        sum%=MODN;
    }
    cout<<sum<<endl;
	return 0;
}

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值