做到这道题目的时候,大脑已经极度不清醒了,头昏昏的,很不舒服。
后来想不能半途而废。要努力做一次完美的比赛。后来就直接做了。
最后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;
}
}
*/