ARC110-D - Binomial Coefficient is Fun(组合数+思维）

题目链接
题意：
给出一个长为n的数组A，一个数组B的和m，问从B中选出b1,b2,…,bn(bi>=ai)，将所有可能的∏(n,i=1) C(bi,ai)相加答案是多少。
思路：
可以先举出几个样例，细心观察会发现答案等于C(sum+n,m+n),当然需要一些对组合数字的敏感度。
然后可以把问题看作用木棍分隔小球的场景。添加木棍的目的是在选取过程中分歌小球。对于A，sum个小球由n个木棍分隔开，小球首尾都存在木棍；对于B，m个小球由n个木棍分隔开。要从B中选出A，即为C(sum+n,m+n)。
至于为什么是n根木棍不是n-1根，最后一根木棍的添加是为了分隔B中m-sum个多余的小球。
代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int N=2e3+5;
const int mod=1e9+7;
int a[N];
ll sum;

ll power(ll a,ll b){
    ll ans=1;
    while(b){
        if(b&1){
            ans=ans*a%mod;
        }
        a=a*a%mod;
        b>>=1;
    }
    return ans%mod;
}

ll C(ll n,ll m){
    ll ans=1;
    n=min(n,m-n);
    for(int i=1;i<=n;i++){
        ll res=power(i,mod-2);
        ans=ans*res%mod;
    }
    for(ll i=m-n+1;i<=m;i++){
        ans=ans*i%mod;
    }
    return ans;
}

int main(){
    ll n,m;cin>>n>>m;
    for(ll i=0;i<n;i++){
        cin>>a[i];
        sum+=a[i];
    }
    if(sum>m){
        cout<<"0"<<endl;
    }
    else cout<<C(sum+n,n+m)%mod<<endl;
    return 0;
}