题目意思:有n件物品,每件物品有两个属性s, f。要求从中选择一些物品,使得它们的 s 值的和 ts 大于零,f 值的和 tf 大于0,并且使得 ts 加上 tf 最大。
关键1:要将问题转化成0-1背包
这是取和不取的问题,很容易想到背包。可以将每件物品的 s 值看成重量,f 值看成价值。定义状态 dp[j] 表示容量为 j 的背包能被装满的情况下能取得的最大价值,那么 j + dp[j] 值最大的那个就是答案了。根据状态的定义,当我们要计算 dp[j] 的时候,如果发现容量为 j 的背包装不满,就可以跳过。
关键2:牛的重量可能为负
也就是背包的容量可能为负数。假设现在背包容量的范围是 [-100000, 100000],我们加上一个偏移量 shift = 100000,使得区间变成 [0, 200000]。shift的取值要比最小的负数的绝对值还要大,这样才能保证把负数映射成正数。
关键3:牛的重量分正负两种情况处理
0-1背包中的循环如下:
for (int j = maxv - 1; j >= s[i]; j--)
{
dp[j] = max(dp[j], dp[j - s[i]] + f[i]);
}
还记得为什么要这么写吗?因为我们要保证 dp[j - s[i]] 是上一次状态的数据。因为 s[i] 大于 0,所以 j - s[i] < j,也就是当我们在计算 j 位置的数据的时候, j 位置左边的数据还未更新。所以我们在计算 dp 数组的时候从右往左。
那么当 s[i] 为负的时候呢?j - s[i] > j,也就是当我们在计算 j 位置的数据的时候, j 位置右边的数据还未更新。所以我们在计算 dp 数组的时候从左往右。
代码如下:
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
const int shift = 100000;
const int maxv = 300000;
const int INF = 0x3f3f3f;
typedef long long int ll;
int main()
{
int n, s[105], f[105], dp[maxv];
scanf("%d", &n);
for (int i = 0; i < n; i++)
scanf("%d%d", &s[i], &f[i]);
for (int i = 0; i < maxv; i++)
dp[i] = (i == shift) ? 0 : -INF; // dp[j]的值等于-INF表示容量为j的背包装不满,大于-INF表示可以装满
for (int i = 0; i < n; i++)
{
if (s[i] >= 0)
{
for (int j = maxv - 1; j >= s[i]; j--)
{
if (dp[j - s[i]] > -INF) // 容量为 j 的背包可以装满
dp[j] = max(dp[j], dp[j - s[i]] + f[i]);
}
}
else
{
for (int j = s[i]; j < maxv + s[i]; j++)
{
if (dp[j - s[i]] > -INF)
dp[j] = max(dp[j], dp[j - s[i]] + f[i]);
}
}
}
int ans = INT_MIN;
for (int i = shift; i < maxv; i++)
{
int t = dp[i] + i - shift;
if (dp[i] >= 0 && t > ans)
ans = t;
}
printf("%d\n", ans);
return 0;
}