今天先来实现一下昨天学的背包,如果还有时间的话,再把之前的算法复习一下~
2
优化后版本
N = 1010
v = [0]*N
w = [0]*N
f = [0]*N
n,m = map(int, input().split())
for i in range(1,1+n):
v[i],w[i] = map(int,input().split())
for i in range(1,n+1):
for j in range(m,v[i]-1,-1):
f[j] = max(f[j],f[j-v[i]]+w[i])
print(f[m])
完全背包问题
每个物体不限制选择的个数,与01背包的区别就是公式由i-1变为了i
朴素法:
N = 1010
v = [0]*N
w = [0]*N
f = [[0]*N for _ in range(N)]
n,m = map(int,input().split())
for i in range(1,1+n):
v[i],w[i] = map(int,input().split())
for i in range(1,1+n):
for j in range(1+m):
f[i][j] = f[i-1][j]
if j>=v[i]:
f[i][j] = max(f[i][j],f[i][j-v[i]]+w[i])
print(f[i][j])
优化后:
不懂为何这么优化,但是知道如何优化
N = 1010
v = [0]*N
w = [0]*N
f = [0]*N
n,m = map(int,input().split())
for i in range(1,1+n):
v[i],w[i] = map(int,input().split())
for i in range(1,1+n):
for j in range(v[i],1+m):
f[j] = max(f[j],f[j-v[i]]+w[i])
print(f[m])
多重背包
朴素版本
# 朴素法不加优化
N = 110
v = [0]*N
w = [0]*N
s = [0]*N
f = [[0]*N for _ in range(N)]
n,m =map(int,input().split())
for i in range(1,n+1):
v[i],w[i],s[i] = map(int,input().split())
for i in range(1,1+n):
for j in range(1+m):
k = 0
f[i][j] = f[i-1][j]
while 1:
if k<=s[i] and k*v[i]<=j:
f[i][j] = max(f[i][j],f[i-1][j-k*v[i]]+k*w[i])
k+=1
else:
break
print(f[n][m])
多重背包-二进制优化
化为01背包问题
N = 11010
M = 2010
v = [0]*N
w = [0]*N
f = [0]*N
n,m = map(int,input().split())
idx = 0
for _ in range(n):
a,b,s = map(int,input().split())
k = 1
while k<=s:
idx += 1
v[idx] = a*k
w[idx] = b*k
s -= k
k*=2
if s>0:
idx += 1
v[idx] = a*s
w[idx] = b*s
n = idx
for i in range(1,1+n):
for j in range(m,v[i]-1,-1):
f[j] = max(f[j],f[j-v[i]]+w[i])
j -= 1
print(f[m])
分组背包
分组背包问题相当于每组只能选一个或者不选,然后求最大值。
由于是用的i-1的值,也就是没更新的,需要从大到小;如果用的是i的值,那么边需要从小到大遍历j,因为j-v【i,k】比j小,从小到大遍历后,再用到j-v【i,k】时就相当于用的是第i层的。也就是先遍历了j,然后才到j-v【i】,相当于把f[j-v[i,k]]更新了,f[j-v[i,k]]就代表着i层的;而从大到小遍历j的话,用的则是尚未更新的i-1层,因为j比j-v[i,k]大,j-v【i,k】还未更新。
总之,要是比较的是i-1层,则从大到小遍历j;
如果用的是i,则从小到大遍历j
这道题,存储v\w\s用的是二维数组,也值得学习,感觉自己想不出来的。
N = 110
v = [[0]*N for _ in range(N)]
w = [[0]*N for _ in range(N)]
s = [[0]*N for _ in range(N)]
f = [0]*N
n,m = map(int,input().split())
# 这样的存储方式值得好好琢磨
for i in range(1,1+n):
s[i] = int(input())
for j in range(s[i]):
v[i][j],w[i][j] = map(int,input().split())
for i in range(1,1+n):
for j in range(m,-1,-1):
# k=0时相当于不选第i组的
for k in range(s[i]):
if v[i][k]<=j:
f[j] = max(f[j],f[j-v[i][k]]+w[i][k])
print(f[m])