砝码称重——空间优化01背包

空间优化01背包

在之前的文章中我们采用二维数组去存，但实际上可以通过一维数组存。

首先，我采用了另一种方式去存背包，这是动态转移方程
flag[i][j] = flag[i - 1][j] + flag[i - 1][j - weight[i]];

其中代表的是前i个砝码达到重量为j的状态共有flag[i][j]种方法。

我个人的想法是把二维数组的最后结果都移动在一起了，可以脑补一个格子图，把最终的结果全部填在第一行的感觉。就是说用后面的状态去覆盖前面的状态。

这样要求我们是倒着存的，因为如果正着存，前面的状态变了会影响后面（因为是加法，所谓的最终量，并不是过程量，而加法中需要过程量）。关于为什么j从v到0的理解放在了这里

#include <stdio.h>
int flag[1100] = {0};
int weight[1100] = {0};
int main()
{
    int cnt = 0;
    int a, b, c, d, e, f;

    scanf("%d%d%d%d%d%d", &a, &b, &c, &d, &e, &f);
    int n = a + b + c + d + e + f;
    int max_weight = 0;
    flag[0] = 1;
    for (int i = 1; i <= n; i++)
    {
        if (i <= a)
            weight[i] = 1;
        else if (i <= (a + b))
            weight[i] = 2;
        else if (i <= (a + b + c))
            weight[i] = 3;
        else if (i <= (a + b + c + d))
            weight[i] = 5;
        else if (i <= (a + b + c + d + e))
            weight[i] = 10;
        else
            weight[i] = 20;
        max_weight += weight[i];
    }

    
    for (int i = 1; i <= n; i++)
    { // 前n个物体
        for (int j = max_weight; j >= 1; j--)
        { 
            if (j >= weight[i])
            {
                flag[j] = flag[j] + flag[j - weight[i]];
            }
        }
    }

    for(int j = 1; j <= max_weight; j++){
        if(flag[j] != 0)
            cnt ++;
    }
    printf("Total=%d", cnt);
}