题目链接:Pick The Sticks - HDU 5543 - Virtual Judge (vjudge.net)
题目大意:你有一条L长的管子,选择n种长度的金条将其放入管中,每根金条都有它的对应的价值,用最好的方法使能选取的金条价值最高,两端允许插入中心位于管中的金条。
思路:01背包的改版,需要考虑金条放入管的两端时的状态。
#include <bits/stdc++.h>
#define int long long
#define dbug(x) cout<<#x<<'='<<x<<endl;
#define x first
#define y second
#define pb push_back
const int N = 2e5 + 5;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + 7;
using namespace std;
typedef pair<int, int>pii;
int n, m;
map<int, int>mp;
int a[N],b[N],dp[N][3];
struct node{
int w,v;
}g[N];
void solve (int u) {
cin>>n>>m;int ans=0; memset(dp, 0, sizeof(dp));//初始化
for(int i=1;i<=n;i++) {
cin>>g[i].w>>g[i].v;
ans=max(ans,g[i].v);//包含出现只放一根金条便可大于其他所有摆放方法的情况
g[i].w*=2;//为防止奇数除以2变小数所以乘2变偶数
}
m*=2;//同上
for(int i=1;i<=n;i++){
for(int j=m;j>=g[i].w/2;j--){//01背包
for(int k=0;k<=2;k++){//k的大小为两端放了几个金条
if(j>=g[i].w)dp[j][k]=max(dp[j][k],dp[j-g[i].w][k]+g[i].v);//第i根金条放到中间
if(k){
dp[j][k]=max(dp[j][k],dp[j-g[i].w/2][k-1]+g[i].v);//放到两边
}
ans=max(ans,dp[j][k]);
}
}
}
printf("Case #%d: %lld\n",u , ans);
}
signed main () {
int T = 1;
std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
cin>>T;
for(int i=1;i<=T;i++) solve (i);
return 0;
}
代码: