【阈值优化+背包】51Nod1597[有限背包计数问题]题解

题目概述

有一个容量为 $n$ 的背包，还有 $n$ 个物品，第 $i$ 个物品体积为 $i$ 且有 $i$ 个。

解题报告

双倍经验题LOJ6089。

可怕的阈值优化……令 $S=\lfloor \sqrt n\rfloor$ ：

• 当物品均 $\le S$ 的时候是系数背包， $f[i][j]=\sum f[i-1][j-ki]$ ，用前缀和优化，记录 $sum[i]=\sum f[i-1][j\ mod\ i]$ 。

• 当物品均 $>S$ 的时候是无限背包，由于物品无限（ $i*i>n$ ）且体积是连续的，所以可以改变定义，用 $g[i][j]$ 表示用了 $i$ 种物品，背包大小为 $j$ 的方案数，我们可以这样来累计方案：

  • $g[i][j-i]$ ，表示把所有物品的体积都加上 $1$ 。
  • $g[i-1][j-S-1]$ ，表示新增一个物品，物品体积为 $S+1$ 。

  由于物品体积均 $>S$ ，所以最多 $S$ 种物品。

最后 $ans=\sum_{i=1}^{S}\sum_{j=0}^{n}f[S][j]\times g[i][n-j]$ 。

示例程序

#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
typedef long long LL;
const int maxn=100000,maxs=316,MOD=23333333;

int n,S,f[2][maxn+5],g[2][maxn+5],sum[maxn+5];

inline void AMOD(int &x,int tem) {if ((x+=tem)>=MOD) x-=MOD;}
int DP()
{
    f[0][0]=1;
    for (int i=1,c=1;i<=S;i++,c^=1)
    {
        memset(sum,0,sizeof(sum));
        for (int j=0;j<=n;j++)
        {
            AMOD(f[c][j]=f[c^1][j],sum[j%i]);AMOD(sum[j%i],f[c^1][j]);
            if (j>=i*i) AMOD(sum[j%i],MOD-f[c^1][j-i*i]);
        }
    }
    g[0][0]=1;int ans=f[S&1][n];
    for (int i=1,c=1;i<=S;i++,c^=1)
    {
        for (int j=0;j<i;j++) g[c][j]=0;
        for (int j=i;j<=n;j++)
            {g[c][j]=g[c][j-i];if (j>S) AMOD(g[c][j],g[c^1][j-S-1]);}
        for (int j=0;j<=n;j++) AMOD(ans,(LL)f[S&1][j]*g[c][n-j]%MOD);
    }
    return ans;
}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    scanf("%d",&n);S=sqrt(n);
    return printf("%d\n",DP()),0;
}