cf554题意的理解和组合问题

来源:互联网 发布:网络安全教育讲话稿 编辑:程序博客网 时间:2024/05/01 20:20

这题当时做的时候不是很理解题意,就撸了一发卢卡斯定理,结果不对,后来理解了好久还是不理解那句话,

百度了题解,发现是第i中颜色的最后一个后面必须是i+1种颜色,然后剩下的就是组合问题,

#include<iostream>#include<cstdio>#include<algorithm>#include<set>#include<string>#define LL long longusing namespace std;LL c[1005][1005];const LL mod=1000000007;int main(){   c[0][0]=1;   for(int i=1;i<=1000;i++)   {       c[i][0]=1;       for(int j=1;j<=i;j++)        c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;   }   int k;   while(cin>>k)   {       int a[1005];       for(int i=1;i<=k;i++)        cin>>a[i];       LL res=1;       int total=0;       for(int i=1;i<=k;i++)       {           res=res*c[total+a[i]-1][a[i]-1]%mod;           total+=a[i];       }       cout<<res<<endl;   }   return 0;}


0 0
原创粉丝点击