题意:现有n头牛,每头牛各有两个指数:聪明指数S和幸福指数F,如果存在S的和TS>=0与F的和TF>=0同时成立时,则输出TS与TF的和的最大值sum,否则输出0。
分析: 取与不取问题很容易想到背包求解,由于题目中有负数,不是一般背包问题,需要进行一定转换将其转变成一般背包,如何将负数下标转换成正数呢?显然需要转移原点位置到100000,开辟维度为200000的数组,开始的位置在100000处 ,我们用dp[i]表示当TS为i时右边TF最大值,这样就很巧妙的转化成01背包问题了,就可以根据s[i]的取值不同采取两种不同的01背包取法。
状态转移公式:dp[i+S]=max(dp[i+S],dp[i]+F)
代码:
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 200005;
const int v = -10000000;
int dp[N];
int left_v, right_v;
int main() {
int n;
int s, f;
left_v = right_v = 100000;
memset(dp, v, sizeof(dp));
dp[left_v] = 0;
cin >> n;
for (int i = 0; i < n; ++i) {
cin >> s >> f;
if (s > 0) {
for (int j = right_v; j >= left_v; --j) {
dp[j + s] = max(dp[j + s], dp[j] + f);
}
right_v += s;
}
else {
for (int j = left_v; j <= right_v; ++j) {
dp[j + s] = max(dp[j + s], dp[j] + f);
}
left_v += s;
}
}
int max_v = 0;
for (int i = 100000; i <= right_v; ++i) {
if (dp[i] >= 0) {
max_v = max(max_v, i + dp[i] - 100000);
}
}
cout << max_v << endl;
return 0;
}