TC-SRM-609-DIV2-950- VocaloidsAndSongs

做到这道题目的时候,大脑已经极度不清醒了,头昏昏的,很不舒服。

后来想不能半途而废。要努力做一次完美的比赛。后来就直接做了。

最后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;
    }
}
*/


你可能感兴趣的:(TC-SRM-609-DIV2-950- VocaloidsAndSongs)