背包问题 AcWing 2. 01背包问题
原题链接
算法标签
背包问题 DP
思路
代码
#include<bits/stdc++.h>
#define int long long
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>b;--i)
using namespace std;
const int N = 10005;
int f[N][N];
int a[N], b[N];
inline int read(){
int s=0,w=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
return s*w;
}
void put(int x) {
if(x<0) putchar('-'),x=-x;
if(x>=10) put(x/10);
putchar(x%10^48);
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n=read(), m=read();
rep(i, 1, n+1){
a[i]=read(), b[i]=read();
}
rep(i, 1, n+1){
rep(j, 0, m+1){
f[i][j]=f[i-1][j];
if(j>=a[i]){
f[i][j]=max(f[i][j], f[i-1][j-a[i]]+b[i]);
}
}
}
int ans=0;
rep(i, 0, m+1){
ans=max(ans, f[n][i]);
}
printf("%lld", ans);
return 0;
}
优化 将状态表示由二维优化为一维
图片转自该题解
证明
f[k]在状态转移过程中更新为背包容量不超过j的最优解
图片转自该视频
若需使f[k]表示背包容量恰为j的最优解,需初始化
f
[
0
]
=
0
,
f
[
i
]
=
−
I
N
F
f[0]=0, f[i]=-INF
f[0]=0,f[i]=−INF即可
图片转自该视频
代码
#include<bits/stdc++.h>
#define int long long
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>b;--i)
using namespace std;
const int N = 10005;
int f[N];
int a[N], b[N];
inline int read(){
int s=0,w=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
return s*w;
}
void put(int x) {
if(x<0) putchar('-'),x=-x;
if(x>=10) put(x/10);
putchar(x%10^48);
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n=read(), m=read();
rep(i, 1, n+1){
a[i]=read(), b[i]=read();
}
rep(i, 1, n+1){
Rep(j, m, a[i]-1){
f[j]=max(f[j], f[j-a[i]]+b[i]);
}
}
printf("%lld", f[m]);
return 0;
}
原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈