题目大意:每头牛有智商和有趣2个值,给你n只牛要求怎么样选择使智商值总和、有趣值总和不为负数并且总和最大。
解题思路:对于每头牛有选和不选2种选择,自然想到01背包,用dp[i]表示有趣值得大小,i表示智商值大小,但是智商值可能为负数,而01背包dp[i]中i为非负数,无法表示。但是,通过看题目的数据范围,n个牛总和无论是智商值还是有趣值都是不超过100000的,那么我们可以设置一个大小为200000+的数组,0~100000表示智商为负数,100000+表示智商为非负数。
代码:
#include <iostream>
#include <stdio.h>
using namespace std;
const int inf=0x3f3f3f3f;
#define mv 200000
int s[110],f[110];
int dp[200010];
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d%d",&s[i],&f[i]);
}
for(int i=0;i<=200000;i++){
dp[i]=-inf;
}
dp[100000]=0;
for(int i=1;i<=n;i++){
if(s[i]>=0){
for(int j=200000;j>=s[i];j--){
dp[j]=max(dp[j],dp[j-s[i]]+f[i]);
}
}else{
for(int j=0;j<=200000+s[i];j++){
dp[j]=max(dp[j],dp[j-s[i]]+f[i]);
}
}
}
int sum=0;
for(int i=100000;i<=200000;i++){
if(dp[i]>=0){
sum=max(sum,dp[i]+i-100000);
//cout<<sum<<endl;
}
}
printf("%d",sum);
return 0;
}