AtCoder Regular Contest 110 D - Binomial Coefficient is Fun

https://atcoder.jp/contests/arc110/tasks/arc110_d

C(n+m,suma+n),suma+n只有4e6

这个问题等价于要从B1中选出A1个，B2中选出A2个，....，Bn中选出An个，且这些简单的选出需要两两之间独立，才会是乘起来

再考虑既然要这个乘积>0才会算进答案，那么必然B1>=A1,B2>=A2....

那么我们就相当于选出suma+n个棒子，前A1个算坐第一个问题中的棒子，第a1+1个为分割棒，把第一个取出问题和第二个取出问题分割了，第A1+2个到A1+1+A2个棒子为第二个问题中的棒子，第A1+1+A2+1个分割棒吧第2个问题和第3个问题分开

然后第suma+n个棒子吧第n个取出问题和多余的小球分开来

那么小球一共就只有m-suma个，小球加棒子一共就是m+n个位置，选出suma+n个放棒子剩下位置放小球的方案数就等价于问题所求

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxl=3e5+10;
const int mod=1e9+7;

int n,m,k,cnt,tot,cas,ans;
int a[maxl];
bool vis[maxl];
char s[maxl];

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

inline ll c(ll n,ll r)
{
	if(r>n || r<0) return 0;
	ll fz=1,fm=1;
	for(int i=1;i<=r;i++)
		fz=fz*(n-i+1)%mod,fm=fm*i%mod;
	return fz*qp(fm,mod-2)%mod; 
}

inline void prework()
{
	ll sum=0;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]),sum+=a[i];
	printf("%lld\n",c(n+m,sum+n));
}

inline void mainwork()
{
	
}

inline void print()
{
	
}

int main()
{
	int t=1;
	//scanf("%d",&t);
	for(cas=1;cas<=t;cas++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}