八级c++GESP:奖品分配

班上有 N 名同学,学号从0 到 N-1 。有M 种奖品要分给这些同学,其中,第 i 种奖品总共有ai 个(i=0,1,…,M-1)。巧合的是,奖品的数量不多不少,每位同学都可以恰好分到一个奖品,且最后剩余的奖品不超过1 个(即:N<=a0+a1+…+aM-1 <=N+1)。
现在,请你求出每个班级礼物分配的方案数,所谓方案,指的是为每位同学都分配一个种类的奖品。只要有一位同学获得了不同种类的奖品,即视为不同的方案。方便起见,你只需要输出方案数对109+7 取模后的结果即可。
共有T 个班级都面临着奖品分配的问题,你需要依次为他们解答。

输入描述

第一行一个整数 ,表示班级数量。
接下来T 行,每行若干用单个空格隔开的正整数。首先是两个正整数N,M ,接着是M 个正整数a0+a1+…+aM-1 。
保证N<=a0+a1+…+aM-1 <=N+1 。

输出描述

输出T 行,每行一个整数,表示该班级分配奖品的方案数对109+7 取模的结果。

用例输入 1 

3
3 2 1 2
3 2 1 3
5 3 3 1 1

用例输出 1 

3
4
20

用例输入 2 

5
100 1 100
100 1 101
20 2 12 8
123 4 80 20 21 3
999 5 101 234 499 66 99

用例输出 2 

1
1
125970
895031741
307187590

提示

在常规程序中,输入、输出时提供提示是好习惯。但在本场考试中,由于系统限定,请不要在输入、输出中附带任何提示信息。

样例解释 1

对于第 1 个班级,学号为0,1,2 的同学可以依次分别获得奖品0,1,1 ,也可以依次分别获得奖品1,0,1 ,也可以依次分别获得奖品1,1,0 ,因此共有3种方案。
对于第 2 个班级,学号为0,1,2 的同学可以依次分别获得奖品0,1,1 ,也可以依次分别获得奖品1,0,1 ,也可以依次分别获得奖品1,1,0 ,也可以依次分别获得奖品1,1,1 ,因此共有4 种方案。
对于第 3 个班级,可以把编号为1 的奖品分配给5 名同学中的任意一名,共有5 种方案;再把编号为2 的奖品分配给剩余4 名同学中的任意一名,共有4 种方案;最后给剩余3 名同学自然获得0 号奖品。因此,方案数为5*4=20。

数据范围

对于30%的测试点,保证N<=10 。
对于另外30%的测试点,保证M=2 。
对于所有测试点,保证N<=1000 ;保证T<=1000 ;保证M<=1001 。

来源

GESP八级

#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
const int mod = 1e9 + 7;
int a[N], C[N][N];
void init() {
    for (int i = 0; i < N; i++) {
        for (int j = 0; j <= i; j++) {
            if (j == 0 || j == i) C[i][j] = 1;
            else C[i][j] = (C[i - 1][j] + C[i - 1][j - 1]) % mod;
        }
    }
}
int main() {
    init();
    int T;
    scanf("%d", &T);
    while(T--) {
        int n, m;
        scanf("%d%d", &n, &m);
        int sum = 0;
        for (int i = 1; i <= m; i++) {
            scanf("%d", &a[i]);
            sum += a[i];
        }
        long long ans = 1;
        for (int i = 1; i <= m; i++) {
            ans = (ans * C[sum][a[i]]) % mod;
            sum -= a[i];
        }
        printf("%lld\n", ans);
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值