集合覆盖问题python

问题描述：

        一个集合U以及U内元素构成的若干个小类集合S={s1,s2,...,sm},每一个si均有一个价值。给的那个一个算法，要找到S的一个子集，该子集满足所含元素包含了所有的元素且使得小类集合的总价值最小。

        例如，U=【1，2，3，4，5】，S=【s1,s2,s3】.其中，s1=【4，1，3】，s2=【2，5】，s3=【1,4,2,3】，并且，Cost（s1）=5,Cost（s2）=10,Cost（s3）=3。选择集合【s2,s3】可完成价值最小的完美覆盖，此时价值为13。

分析：

要完成价值最小，则选择小类集合的条件是价值最小优先。故按照价值的大小去选择，然后判断小类集合是否存在集合U中

def get_mincost(S,U,cost):
    final=[]#存储最后选择的小类集合
    temp=sorted(cost,reverse=False)#保持原来数组顺序不变，方便在S中找到对应的一个小类集合，从小到大排序
    costs = 0#完成覆盖的价值
    for i in range(len(temp)):
        min=temp[i]
        flag = 0#判断S中小类集合是否有元素存在U中
        for j in range(len(cost)):
            if cost[j]==min:
                print(j)
                for k in S[j]:
                    if (k in U) and (len(U)>0):
                        flag=1
                        U.remove(k)#将集合中存在的元素从U中删除
                    if len(U) == 0:
                        break
                if flag==1:#小类集合中有元素存在于U中，则存入final，并且价值算到总价值中
                    costs+=cost[j]
                    final.append(S[j])
    return final,costs
S=[[4,1,3],[2,5],[1,4,3,2]];U=[1,2,3,4,5]
cost=[5,10,3]
final,costs=get_mincost(S,U,cost)
print(final,costs)