算法分析
逆向 dp
从前
个瓜中取出总质量为
的瓜的方案数
滚动数组优化过后的 表示的是总质量为
的方案数
已知各种重量的方案数,求这些若干个西瓜的重量是多少。等于就是将背包中的物品给取出来,询问物品的信息。
那么我们先看看正向是怎么做的
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]
这样子就成功的将一个瓜给取出来了。
那么按照直线思维 是未知数,怎么转移咧
那么没有重量,我们就枚举呗,反正总质量最多不超过
那么思路就逐渐明晰起来了,枚举一下瓜的质量,枚举一下当前取出这个瓜后的质量,就可以得到
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;
}