bzoj2287

背包+fft

既然要不选一个东西，那么我们求出前缀背包和后缀背包，每次答案就是f[i-1][w]*g[i+1][j-w]

但是这样复杂度还是n^3，跑不过，但是我们发现上面那个东西不就是个裸卷积吗，直接上fft，但是wa了...

wa的程序，大概是精度问题吧

#include<bits/stdc++.h>
using namespace std;
#define pi acos(-1)
const int N = ;
int n, m, L, x, nn, mm;
int r[N * ], f[N][N], g[N][N], w[N];
complex<double> a[N * ], b[N * ];
void fft(complex<double> *a, int f)
{
    for(int i = ; i < n; ++i) if(i < r[i]) swap(a[i], a[r[i]]);
    for(int i = ; i < n; i <<= )
    {
        complex<double> t(cos(pi / i), f * sin(pi / i));
        for(int p = i << , j = ; j < n; j += p)
        {
            complex<double> w(, );
            for(int k = ; k < i; ++k, w *= t)
            {
                complex<double> x = a[j + k], y = w * a[j + k + i];
                a[j + k] = x + y; a[j + k + i] = x - y;
            }
        }
    }
}
int main()
{
    scanf("%d%d", &nn, &mm);
    for(int i = ; i <= nn; ++i) scanf("%d", &w[i]);
    f[][] = ;
    for(int i = ; i <= nn; ++i)
        for(int j = ; j <= mm; ++j)
        {
            f[i][j] = f[i - ][j];
            if(j >= w[i]) f[i][j] = (f[i][j] + f[i - ][j - w[i]]) % ;
        }
    g[nn + ][] = ;
    for(int i = nn; i; --i)
        for(int j = ; j <= mm; ++j)
        {
            g[i][j] = g[i + ][j];
            if(j >= w[i]) g[i][j] = (g[i][j] + g[i + ][j - w[i]]) % ;
        }
    for(int i = ; i <= nn; ++i)
    {
        for(int j = ; j <= mm; ++j)
        {
            L = ;
            m =  * j + ;
            for(int k = ; k <= m; ++k) a[k] = b[k] = ;
            for(int k = ; k <= j; ++k)
            {
                a[k] = f[i - ][k];
                b[k] = g[i + ][k];
            }
            for(n = ; n <= m; n <<= ) ++L;
            for(int k = ; k < n; ++k) r[k] = (r[k >> ] >> ) | ((k & ) << (L - ));
            fft(a, );
            fft(b, );
            for(int k = ; k <= n; ++k) a[k] = a[k] * b[k];
            fft(a, -);
            int ans = (int)(a[j].real() / (double)n + 0.5);
            printf("%d", ans % );
        }
        puts("");
    }
    return ;
}

写了一个正解

f[i]:装满i的方案数

c[i][j]:装满j不用i的方案数

j<w[i],自然c[i][j]=f[j],因为w[i]装不下，不可能选

j>=w[i],c[i][j]=f[j]-c[i][j-w[i]],在j-w[i]填上一个w[i]就是j，表示选到第i个物品一定选了i的方案数，相减就是不选的方案数

#include<bits/stdc++.h>
using namespace std;
const int N = ;
int n, m;
int w[N], c[N], f[N];
int main()
{
    scanf("%d%d", &n, &m);
    for(int i = ; i <= n; ++i) scanf("%d", &w[i]);
    f[] = ;
    for(int i = ; i <= n; ++i)
        for(int j = m; j >= w[i]; --j)
            f[j] = (f[j] + f[j - w[i]]) % ;
    for(int i = ; i <= n; ++i)
    {
        for(int j = ; j <= m; ++j)
        {
            if(j >= w[i]) c[j] = (f[j] - c[j - w[i]] %  + ) % ;
            else c[j] = f[j];
            if(j > ) printf("%d", c[j]);
        }
        puts("");
    }
    return ;
}