week 8的内容还是dp,难度上又加大了。
疯狂的采药
这是一道完全背包。状态转移方程式是f[i-1][j]和f[i-1][j-v[i]]+c[i] 。前者表示不选,后者表示选。因为要保证方程的式子左边未知,右边全部已知。所以在循环的时候,空间的枚举范围是v[i]小于等于j小于等于m。(数据要开longlong)
代码:
#include<bits/stdc++.h>
using namespace std;
const int maxm = 10010, maxt = 10000010;
long long v[maxm], t[maxm], f[maxt];//开longlong!
int main(){
int T , m;
cin >> T >> m;
for(int i = 1;i <= m ;i ++) cin >> t[i] >> v[i];
for(int i = 1;i <= m;i ++){
for(int j = t[i];j <= T;j ++){
f[j] = max(f[j],f[j - t[i]] + v[i]);
}
}
cout << f[T];
}
樱花
通过这道题了解了一下多重背包。这道题就是01背包、多重背包和完全背包的混合。在进行到第i棵树的时候,对其进行判断属于哪一种背包。
代码:
#include<bits/stdc++.h>
#define re register int
using namespace std;
int nx,ny,ex,ey,n,f[1010];
int a[10005],b[10005],c[10005];
int tp,co[1000005],v[1000005];//尽可能开大
inline void pre() {
for(re i=1;i<=n;i++) {
int t=1;
while(c[i]) {
co[++tp]=t*a[i];
v[tp]=t*b[i];
c[i]-=t; t*=2;
if(c[i]<t) {//如果剩下的不能再拆,就直接放在一起
co[++tp]=a[i]*c[i];
v[tp]=b[i]*c[i];
break;
}
}
}
}
int main() {
scanf("%d:%d%d:%d%d",&nx,&ny,&ex,&ey,&n);
int t=(ex*60+ey)-(nx*60+ny);
for(re i=1;i<=n;i++) {
scanf("%d%d%d",&a[i],&b[i],&c[i]);
if(!c[i]) c[i]=999999;
}
pre();//二进制拆分
for(re i=1;i<=tp;i++) {//考虑每个拆出来的物品
for(re j=t;j>=co[i];j--)//01背包
f[j]=max(f[j],f[j-co[i]]+v[i]);
}
printf("%d",f[t]);
return 0;
}
摆花
直接压成一维的01背包。
代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn=105, mod = 1000007;
int n, m, a[maxn], f[maxn][maxn];
int main()
{
cin>>n>>m;
for(int i=1; i<=n; i++) cin>>a[i];
f[0][0] = 1;
for(int i=1; i<=n; i++)
for(int j=0; j<=m; j++)
for(int k=0; k<=min(j, a[i]); k++)
f[i][j] = (f[i][j] + f[i-1][j-k])%mod;
cout<<f[n][m]<<endl;
return 0;
}
金明的预算方案
这道题比01背包多了几个内容,通过对比能够更直观的表现出来。
01:选这个东西放包里,不要这个东西
现:不买主件,买主件,买主件+副件1,买主件+副件2,买主件+副件1+副件2
代码:
#include<bits/stdc++.h>
using namespace std;
const int kMaxN = 32000, kMaxM = 60; // 常量名前带 k,符合命名规范
int v[kMaxM + 5][3], p[kMaxM + 5][3];
int f[kMaxN + 5];
int main() {
int n, m;
std::cin >> n >> m;
for (int i = 1; i <= m; ++i) {
int _v, _p, _q;
std::cin >> _v >> _p >> _q;
if (!_q) { // 是主件
v[i][0] = _v;
p[i][0] = _p;
} else {
if (v[_q][1] == 0) { // 是第一个附件
v[_q][1] = _v;
p[_q][1] = _p;
} else { // 是第二个附件
v[_q][2] = _v;
p[_q][2] = _p;
}
}
}
for (int i = 1; i <= m; ++i)
for (int j = n; j >= 0; --j) {
// 用 lambda 表达式代替函数定义(C++11 偷懒专用)
auto cost2 = [v, p, i](int x, int y) { return v[i][x] + v[i][y]; };
auto cost3 = [v, p, i](int x, int y, int z) { return v[i][x] + v[i][y] + v[i][z]; };
auto rpp = [v, p, i](int x) { return v[i][x] * p[i][x]; };
if (j >= v[i][0]) // 够买主件
f[j] = std::max(f[j], f[j - v[i][0]] + rpp(0));
if (j >= cost2(0, 1)) // 还够买第一个附件
f[j] = std::max(f[j], f[j - cost2(0, 1)] + rpp(0) + rpp(1));
if (j >= cost2(0, 2)) // 还够买第二个附件
f[j] = std::max(f[j], f[j - cost2(0, 2)] + rpp(0) + rpp(2));
if (j >= cost3(0, 1, 2)) // 还够买两个附件
f[j] = std::max(f[j], f[j - cost3(0, 1, 2)] + rpp(0) + rpp(1) + rpp(2));
}
std::cout << f[n] << std::endl;
}