TC-SRM-609-DIV2-950- VocaloidsAndSongs

1853 ワード

この問題をやり遂げる時、脳はすでに極度に目が覚めなくて、頭がぼんやりしていて、とても気分が悪いです.
後で中途半端にしてはいけないと思った.完璧な試合を頑張ります.後で直接やりました.
最後にrank 50は、div 2でしたが、比較的嬉しかったです.
cha段階では、最初は2つのエラーのコードを見ましたが、手が遅すぎて、他の人にchaされました.悲しい...
本題に戻る:
前処理dp(i,j):iからj個をとる取り方.
ans[i]:全部でi曲あり、結果はいくらですか.
i曲は、3種類の人がそれぞれ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;
    }
}
*/