奇怪的分组
题目背景
终于解出了 dm 同学的难题,dm 同学同意帮 v 神联络。可 dm 同学有个习惯,就是联络同学的时候喜欢分组联络,而且分组的方式也很特别,要求第 i i i 组的的人数必须大于他指定的个数 C i C_i Ci。在 dm 同学联络的时候,v 神在想,按照 dm 同学的规则一共可以有多少种方案呢?他想啊想,终于……没想出来。于是他又想到了聪明的你,你能帮 v 神算出按照 dm 同学的规则有多少种分组方案吗?
题目描述
v 神的班级共有 N N N 个人,dm 同学想把同学分成 M M M 组联络,要求第 i i i 组的人数必须大于给定的正整数 C i C_i Ci,求有多少不同的方案(两个是相同的方案当且仅当对于任意的一队 i i i,两个方案的第 i i i组同学数量相等)。由于结果很大,所以你只需要输出模 1 0 9 + 7 10^9+7 109+7 的值。
输入格式
第一行两个整数 N N N 和 M M M,后面有 M M M 行,每行一个整数,表示 C i C_i Ci。
输出格式
仅有一行,一个整数,方案数模 1 0 9 + 7 10^9+7 109+7 的值。
样例 #1
样例输入 #1
10 3
1
2
3
样例输出 #1
3
提示
样例解释
方案有三种,每组的个数分别是 ( 3 , 3 , 4 ) (3,3,4) (3,3,4), ( 2 , 4 , 4 ) (2,4,4) (2,4,4), ( 2 , 3 , 5 ) (2,3,5) (2,3,5)。
数据范围约定
对于 30 % 30\% 30% 的数据, N , M ≤ 10 N,M\le 10 N,M≤10。
对于 60 % 60\% 60% 的数据, N , M ≤ 1000 N,M\le 1000 N,M≤1000。
对于 100 % 100\% 100% 的数据, 1 ≤ N , M ≤ 1 0 6 1\le N ,M\le 10^6 1≤N,M≤106, 1 ≤ C i ≤ 1000 1\le C_i\le 1000 1≤Ci≤1000。
数据保证至少有一个方案。
思路
这道题不就是简单的隔板问题,我们本质上求的就是 C n − s u m − 1 m − 1 C_{n-sum-1}^{m-1} Cn−sum−1m−1,其中 s u m sum sum 表示现有球数。
代码( 1 0 6 10^6 106的数据规模求组合数可以用)
//这道题本质上就是求C(n-1,m-1)
#include<iostream>
#include<algorithm>
#include<cstring>
#define int long long
using namespace std;
const int N = 1e6+10,mod = 1e9+7;
int n,m;
int sum,w[N];
int fac[N],infac[N];
int qmi(int a,int b,int c){
int res=1;
while(b){
if(b&1)res=res*a%c;
a=a*a%c;
b>>=1;
}
return res;
}
int C(int a,int b){//C(a,b)
if(b>a)return 0;
return (fac[a]*infac[b]%mod*infac[a-b]);
}
void init(){
fac[0]=infac[0]=1;
for(int i=1;i<N-10;i++){
fac[i]=fac[i-1]*i%mod;
infac[i]=infac[i-1]*qmi(i,mod-2,mod)%mod;
}
}
signed main(){
cin>>n>>m;
for(int i=1;i<=m;i++){
cin>>w[i];
sum+=w[i];
}
init();
cout<<(C(n-sum-1,m-1))%mod;
return 0;
}