最小划分中体现的背包思想

本博客将用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:

原创，转载请标明