Codeforces Round #568 (Div. 2) G2. Playlist for Polycarp (hard version)

因为不会打公式，随意就先将就一下？

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
typedef long long LL;
const int N=;
const int MOD=1e9+;
int add (int x,int y)   {x=x+y;return x>=MOD?x-MOD:x;}
int mul (int x,int y)   {return (LL)x*y%MOD;}
int dec (int x,int y)   {x=x-y;return x<?x+MOD:x;}
int Pow (int x,int y)
{
    if (y==) return ;
    if (y==) return x;
    int lalal=Pow(x,y>>);
    lalal=mul(lalal,lalal);
    if (y&) lalal=mul(lalal,x);
    return lalal;
}
int n,m;
int g[N][N][N][];
int h[N][N][N];
int JC[N];
void Init (int n)
{
    memset(g,,sizeof(g));
    g[][][][]=;g[][][][]=;g[][][][]=;
    JC[]=;for (int u=;u<=n;u++) JC[u]=mul(JC[u-],u);
    for (int u=;u<=n;u++)
    for (int i=;i<=n;i++)
    for (int j=;j<=n;j++)
    for (int k=;k<=;k++)    if (g[u][i][j][k])
    {
        if (k!=) g[u+][i][j][]=add(g[u+][i][j][],g[u][i][j][k]);
        if (k!=) g[u][i+][j][]=add(g[u][i+][j][],g[u][i][j][k]);
        if (k!=) g[u][i][j+][]=add(g[u][i][j+][],g[u][i][j][k]);
        int now=mul(mul(mul(JC[u],JC[i]),JC[j]),g[u][i][j][k]);
        //if (u==2&&i==1&&j==0) printf("%d %d\n",g[u][i][j][k],(mul(JC[u],JC[i]),JC[j]));
        h[u][i][j]=add(h[u][i][j],now);
    }
    //printf("%d\n",h[2][1][0]);
}
vector<int> vec[N];
int f[][N][];
int sum[];
int d[N][N][];
int main()
{
    Init();
    scanf("%d%d",&n,&m);
    for (int u=;u<=n;u++)
    {
        int t,g;
        scanf("%d%d",&t,&g);
        sum[g-]+=t;vec[g-].push_back(t);
    }
    if (sum[]>sum[])    {swap(sum[],sum[]);swap(vec[],vec[]);}
    if (sum[]>sum[])    {swap(sum[],sum[]);swap(vec[],vec[]);}
    for (int u=;u<=;u++)
    {
        int siz=vec[u].size();
        f[u][][]=;
        for (int i=;i<siz;i++)
        for (int j=i+;j>=;j--)
        for (int k=sum[u];k>=vec[u][i];k--)
        f[u][j][k]=add(f[u][j][k],f[u][j-][k-vec[u][i]]);
    }
    for (int u=;u<=vec[].size();u++)
    for (int j=;j<=sum[];j++) if (f[][u][j])
    for (int i=;i<=vec[].size();i++)
    for (int k=;k<=sum[];k++) if (f[][i][k])
    d[u][i][j+k]=add(d[u][i][j+k],mul(f[][u][j],f[][i][k]));
    int ans=;
    for (int u=;u<=vec[].size();u++)
    for (int i=;i<=sum[];i++) if (f[][u][i])
    {
        int t=m-i;
        for (int j=;j<=vec[].size();j++)
        for (int k=;k<=vec[].size();k++)
        ans=add(ans,mul(h[u][j][k],(mul(d[j][k][t],f[][u][i]))));
    }
    printf("%d\n",ans);
    return ;
}