本博客将用Python 3 实现最小划分求解:给出一个正整数数组,写一个程序把这个整数数组分成S1跟S2两部分,使S1中的和跟S2中的和的绝对值最小。如果有一个一个整数数组 S
有 n
个数,如果Subset1有 m
个数,Subset2必须有 n-m
个数并且 abs(sum(Subset1) – sum(Subset2))
应该最小
这是面试中常见的题型之一,体现的正是背包算法的思想。
#coding=utf-8
'''
Author:ZhangLongQi
Date:2018-4-1
E-mail:zlonqi@foxmail.com
Environment:Python 3.6
Descrip:最小划分,背包思想
Improve:优化实现过程
'''
def bag(nums,cap,wet=[],val=[]):
odd=[]
def prints(path,i,j):
if i>0 and j>0:
if path[i][j]==1:
odd.append(i)
prints(path,i-1,j-wet[i])
else:
prints(path,i-1,j)
res=[[0 for i in range(cap+1)] for j in range(nums+1)]
path=[[0 for i in range(cap+1)] for j in range(nums+1)]
for i in range(1,nums+1):
for j in range(1,cap+1):
res[i][j]=res[i-1][j]
if wet[i]<=j:
tmp= val[i]+res[i-1][j-wet[i]]
if tmp>res[i-1][j]:
res[i][j]=tmp
path[i][j]=1
#for i in range(len(path)):
#print(path[i])
prints(path,nums,cap)
return odd,res[nums][cap]
def dsp(k=[]):
s=0
for i in k:
s=s+w[i]
return abs((sum(w)-s)-s)
if __name__=='__main__':
w=[2,3,1,4,6,5]
print('w=',w)
c1=sum(w)//2
c2=sum(w)//2+1 #仅为了比较
n=len(w)
val=[1 for _ in range(n)]
w.insert(0,0)
val.insert(0,0)
k1,v=bag(n,c1,w,val)
k2,v=bag(n,c2,w,val)
k=k1 if dsp(k1)<dsp(k2) else k2 #Python中没有三目运算符
print('把第',sorted(k),'个数取出,差值为',dsp(k))
Demon:
原创,转载请标明