做到这道题目的时候,大脑已经极度不清醒了,头昏昏的,很不舒服。
后来想不能半途而废。要努力做一次完美的比赛。后来就直接做了。
最后rank50,虽然是在div2,但是也是比较高兴的。
cha阶段,我一开始看见两个错误的代码,但是下手太慢了,就被别人cha掉了。伤心。。。
言归正传:
预处理dp(i,j):从i个中取j个的取法。
ans[i]:一共有i首歌曲,结果是多少。
i首歌曲,三种人分别有a,b,c个。
如果不考虑某一首歌可能没人唱的可能那么:ans[i]=dp[i][a]*dp[i][b]*dp[i][c];
如果一共有j首歌没人唱的可能性一共有:(dp[i][i-j]*ans[j])%MOD;
最终ans[n]就是结果。
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<stdlib.h> #include<vector> using namespace std; #define INF 0xfffffff #define MOD 1000000007 #define LL long long LL dp[101][101]; LL ans[101]; void init() { int i,j; for(i=0;i<51;i++) { for(j=0;j<51;j++) { dp[i][j]=0; } } dp[1][1]=1; for(i=2;i<=51;i++) { dp[i][1]=i; for(j=2;j<=51;j++) { dp[i][j]=(dp[i-1][j-1]+dp[i-1][j])%MOD; } } } int dos(int n,int a,int b,int c) { if(a>b)swap(a,b); if(a>c)swap(a,c); if(b>c)swap(b,c); int all=a+b+c; if(all<n)return 0; init(); int i,j; for(i=1;i<=n;i++) { if(i<c)ans[i]=0; else { ans[i]=(((dp[i][a]*dp[i][b])%MOD)*dp[i][c])%MOD; for(j=i-1;j>=1;j--) { ans[i]-=(dp[i][i-j]*ans[j])%MOD; } while(ans[i]<0)ans[i]+=MOD; } } return ans[n]; } class VocaloidsAndSongs { public: int count(int S, int gumi, int ia, int mayu) { return dos(S,gumi,ia, mayu); } }; /* int main() { int n,a,b,c; while(scanf("%d%d%d%d",&n,&a,&b,&c)) { cout<<dos(n,a,b,c)<<endl; } } */