CQUPT第九届ACM校赛 H 夺命毒奶

1524: 夺命果奶

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 187   Solved: 16
[ Submit][ Status][ Web Board]

Description

CC公主被抓走了，为了救回她，你要通过九个关卡的考验。

在这一关中，你要喝下你调配出的果奶。果奶由名叫果苣的植物加上牛奶搭配而成。你的面前有很多果苣，但是其中有一些是黑暗果苣，如果果奶中有太多种黑暗果苣，那么这杯果奶就变成了毒奶，你就会被奶死。

请问你有多少种配出安全果奶的办法？

Input

输入的第一行包含一个正整数T(1<=T<=100)。T表示数据组数。

每一组数据的第一行包括四个整数N1(1<=N1<=100)、N2(0<=N2<=N1)、M(1<=M<=10)和K(0<=K<=9)。N1表示果苣的种类数，N2表示黑暗果苣的种类数，M表示果奶由M种不同种类的果苣调配而成，K表示当果奶中含有的黑暗果苣不少于K种那么这种配法配出的是毒奶。

Output

对于每组数据，在单独的一行中输出一个整数表示配出安全果奶的办法数。当两种配法选择的M种果苣相同，这两种配法被认为是相同的方法。

Sample Input

4
3 3 1 4
4 2 2 2
100 50 2 2
4 3 2 1

Sample Output

3
5
3725
0

题解：想是很好想的，先求出总的组合情况C(n1,m)，再减去是毒奶的情况。但是最难的就是如何才能不爆long long，比赛的时候想了好久也想不出来。看题解才知道的：C(n,m) = n! / (m! * (n - m)!) = n * (n -1) * (n -2) * .....*(n - m + 1) / m! = n / 1 * (n - 1) / 2 * (n - 2) / 3*.....*(n - m + 1) / m。

我还试了一下这个公式：C(n,m) = C(n - 1,m - 1) + C(n - 1,m)。我用递归做，但是发现到C(100,4)左右就开始变得很慢了，速度不行。

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

const int maxn = 105;

long long f(int n,int m)
{
    long long ans = 1;
    if(n < m)
    {
        return 0;
    }
    for(int i = 1;i <= m;i++)
    {
        ans = ans * n;
        n--;
        ans = ans / i;
    }
    return ans;
}

int main()
{
    int T;
    int n1,n2,m,k;
    long long sum;
    long long num;
    long long ans;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d%d",&n1,&n2,&m,&k);
        sum = f(n1,m);
        num = 0;
        for(int i = k;i <= m;i++)
        {
            num += f(n2,i) * f(n1 - n2,m - i);
        }
        ans = sum - num;
        printf("%lld\n",ans);
    }
    return 0;
}