智乃买瓜(another version)_coat another version-CSDN博客

本文链接：https://blog.csdn.net/m0_56007815/article/details/123015129

这篇博客介绍了如何利用逆向动态规划解决从一组瓜中取出特定总质量的瓜的方案数问题。通过滚动数组优化，博主展示了如何将正向的动态规划状态转移方程改写为逆向形式，并通过枚举瓜的质量来恢复具体的瓜的重量。代码示例展示了详细的逆向DP转移过程，确保方案数的正确性，并给出了最终答案的输出格式。

摘要由CSDN通过智能技术生成

算法分析

逆向 dp

$dp[i][j]:$ 从前 $i$ 个瓜中取出总质量为 $j$ 的瓜的方案数

滚动数组优化过后的 $dp[i]$ 表示的是总质量为 $i$ 的方案数

已知各种重量的方案数,求这些若干个西瓜的重量是多少。等于就是将背包中的物品给取出来,询问物品的信息。

那么我们先看看正向是怎么做的

dp[i][j] = dp[i - 1][j] + dp[i - 1][j - w[i]] + dp[i - 1][j - w[i] / 2]

那么按照正常的逆向思维就是将式子进行移项

dp[i - 1][j] = dp[i][j] - dp[i - 1][j - w[i]] - dp[i - 1][j - w[i] / 2]

这样子就成功的将一个瓜给取出来了。

那么按照直线思维 $w[i]$ 是未知数,怎么转移咧

那么没有重量,我们就枚举呗,反正总质量最多不超过 $1000$

那么思路就逐渐明晰起来了,枚举一下瓜的质量,枚举一下当前取出这个瓜后的质量,就可以得到

for(int i = 1;i <= m;i ++)//枚举要买的瓜的质量
    {
        while(dp[i])//如果可以买 i 重量的瓜
        {
            ans.push_back(i * 2);
            if(ans.size() )
            assert(ans.size() <= 1000);
            for(int j = 0;j <= m;j ++)//当前取出这个瓜后的质量
            {
                if(j + i <= m) dp[j + i] -= dp[j];
                if(j + i <= m && dp[j + i] < 0) dp[j + i] += mod;
                if(j + i + i <= m) dp[j + i + i] -= dp[j];
                if(j + i + i <= m && dp[j + i + i] < 0) dp[j + i + i] += mod;
            }
        }
    }

注意一下,因为方案数都是对 mod 取余过的,因此可能会出现负数,记得将 mod 补上即可

AC code

#include<bits/stdc++.h>
using namespace std;
const int N = 1e3 + 10;
const int mod = 1e9 + 7;

long long dp[N];
vector<int>ans;
// dp[i][j] = dp[i - 1][j] + dp[i - 1][j - w[i]] + dp[i - 1][j - w[i] / 2]
// dp[i - 1][j] = dp[i][j] - dp[i - 1][j - w[i]] - dp[i - 1][j - w[i] / 2]
int main()
{
    int m;
    cin >> m;
    for(int i = 1;i <= m;i ++) cin >> dp[i];
    
    //ex: dp[2] 对于质量为 2 的瓜 可能是在 dp[0] 的基础上买了一整个瓜 
    //也有可能是在 dp[1] 的基础上买了半个瓜
    //因此要将他们的方案数减掉
    dp[0] = 1;//记得初始化
    for(int i = 1;i <= m;i ++)//枚举要买的瓜的质量
    {
        while(dp[i])//如果可以买 i 重量的瓜
        {
            ans.push_back(i * 2);
            if(ans.size() )
            assert(ans.size() <= 1000);
            for(int j = 0;j <= m;j ++)//当前取出这个瓜后的质量
            {
                if(j + i <= m) dp[j + i] -= dp[j];
                if(j + i <= m && dp[j + i] < 0) dp[j + i] += mod;
                if(j + i + i <= m) dp[j + i + i] -= dp[j];
                if(j + i + i <= m && dp[j + i + i] < 0) dp[j + i + i] += mod;
            }
        }
    }
    cout << ans.size() << endl;
    for(int i=0;i<ans.size();++i)
	{
		printf("%d%c",ans[i]," \n"[i+1==ans.size()]);
	}
    return 0;
}