codeforces 553A Kyoya and Colored Balls（组合数学）

题目链接：

http://codeforces.com/problemset/problem/553/A

题目大意：

有n个球，有k种颜色，这k种颜色会涂在这些球上。对于颜色i，最后一个涂这个颜色的球的后面那个球的颜色一定是i+1。

现在问球涂色后的排序有多少种方案。

思路：可以知道涂颜色k的最后一个球一定在最后一个位置上，那么剩下的a[k]-1个球就可以在n-1个位置任意摆放，有C（a[k]-1,n-1）种放法。然后颜色为k-1的最后一个球就摆放在颜色为k球的前面一位，剩下的a[k-1]-1个球就可以在n-a[k]-1的位置上摆放，有C（a[k-1]-1,n-a[k]-1）种放法。

依次类推，直到所有的球都放完。那么答案就是上述的放法相乘然后取模。

但是对于C（m,n），虽然我们可以通过阶乘A（m,n）/A(m,m)计算，但是数据范围一大，long long范围下就存不下了，如果采用取模方式，就会出错。

所以我们想到了一个公式：C（m,n）=C(m,n-1)+C(m-1,n-1)。

那么我们就可以设一个数组来记录下C（m,n）的答案。

代码：

#include<stdio.h>
#include<string.h>
#include<math.h>
#define ll  __int64
#define mod 1000000007
ll c[1005][1005];
int main()
{
    ll k,i,j,n,a[1005];
    for(i=0;i<=1000;i++)
    c[i][0]=1;
    for(i=1;i<=1000;i++)
    {
        for(j=1;j<=i;j++)
        {
            if(i==j)c[i][j]=1;
            else if(i>j)
                c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
        }
    }                                          //记录下1000以内的C(m,n)
    while(scanf("%I64d",&k)!=EOF)
    {
        n=0;
        memset(a,0,sizeof(a));
        for(i=1;i<=k;i++)
      {
          scanf("%I64d",&a[i]);
          n+=a[i];
    }
   // printf("%I64d\n",n);
     ll ans=1;
    for(i=k;i>=1;i--)
    {
       ans=((ans%mod)*(c[n-1][a[i]-1])%mod)%mod;
    //   printf("%I64d %I64d %I64d\n",ans,c[n-1][a[i]-1],n);
       n=n-a[i];
       ans=ans%mod;

    }
    printf("%I64d\n",ans);
    }
    return 0;
}