CodeForces 630 G. Challenge Pennants(组合数学)

Description
公司有5面A锦旗,3面B锦旗,先欲将这些锦旗全部发给n位员工,每位员工可以得到任意种类任意数量的锦旗,问有多少种发放方案
Input
一个整数n表示公司员工数(1<=n<=500)
Output
发放方案数
Sample Input
2
Sample Output
24
Solution
5面A锦旗发放方案为11111,1113,122,113,14,23,5
方案数为cnt1=C(n,5)+4*C(n,4)+2*3*C(n,3)+2*2*C(n,2)+C(n,1)
3面B锦旗发放方案为111,12,3
方案数为cnt2=C(n,3)+2*C(n,2)+C(n,1)
总方案数为ans=cnt1*cnt2
Code

#include<cstdio>
#include<iostream>
using namespace std;
typedef long long ll;
int n;
ll C(int n,int m)
{
    if(m>n)return 0;
    ll ans=1ll;
    if(n-m>m)m=n-m;
    for(int i=1;i<=n-m;i++)
        ans=ans*(m+i)/i;
    return ans;
}
int main()
{
    while(~scanf("%d",&n))
    {
        ll ans=(C(n,1)+4*C(n,2)+6*C(n,3)+4*C(n,4)+C(n,5))*(C(n,1)+2*C(n,2)+C(n,3));
        printf("%I64d\n",ans);
    }
    return 0;
}

你可能感兴趣的:(CodeForces 630 G. Challenge Pennants(组合数学))