找出最大的加权子集,使子集中的任何两个元素的总和不为k。

问题描述 投票:0回答:1

子集的权重是每个键的值的长度。

这是我的尝试。

def nonDivisibleSubset(k, s):
    remdict={}
    result=[]
    for i in range(len(s)):
        rem=s[i]%k
        if rem not in remdict:
            remdict[rem]=[s[i]]
        else:
            remdict[rem].append(s[i])
    b=dict(sorted(remdict.items(), key= lambda x: len(x[1]), reverse=True))


Input: 
k=7
{2: [576, 338, 149, 702, 282, 436], 6: [496, 727, 209], 5: [278, 124], 4: [410, 718], 1: [771, 575]}

从这个字典中 我只想附加2,6,4这几个键的值 因为2+6! =7 2+4! =7 6+4!

python algorithm subset
1个回答
0
投票

所以首先--我们需要找到所有不同的组合,然后从中找到最大的组合。

这是找所有不同组合的代码。

def check(d, s, k):
        for i in d:
            if i + s == k:
                return False
        return True

for i in s:
    temp = [i]
    d = s.copy()
    d.pop(i)
    for j in d:
        if check(temp, j, k):
            temp.append(j)
     temp.sort()
        res_lst.append(temp)
    unique_data = [list(x) for x in set(tuple(x) for x in res_lst)]

(参考如何找到 unique_data: Python: 列表的唯一性)

来找到最大的子集。

res = []
res_sum = 0
for l in unique_data:
    tmp_sum = 0
    for n in l:
        tmp_sum += len(s[n])
    if tmp_sum > res_sum:
        res = l
        res_sum = tmp_sum
return res

0
投票

我们的想法是根据数字除以k后的余数 将s中的数字分组 并将余数存储为key,将frequencycount存储为value. 然后在max(key,k-key)之间选取,如果key==0或者key+key=k,确保只选取一个。在不使用字典的情况下,可能也有其他的方法来解决这个问题。

def nonDivisibleSubset(k, s):
        remdict={}
        result=[]
        keyss=[]
        for i in range(len(s)):
            rem=s[i]%k
            if rem not in remdict:
                remdict[rem]=1
            else:
                remdict[rem]+=1
        #b=dict(sorted(remdict.items(), key= lambda x: len(x[1]), reverse=True))
        print(remdict)
        for key,value in remdict.items():
            if key==0:
                result.append(1)
            elif key+key==k:
                result.append(1)
            elif (k-key) in remdict.keys() and k-key not in keyss :
                print(max(remdict[key],remdict[k-key]))
                keyss.append(key)
                result.append(max(remdict[key],remdict[k-key]))
            elif (k-key) not in remdict.keys():
                print(remdict[key])
                result.append(remdict[key])
        print(result)
        return (sum(result))
© www.soinside.com 2019 - 2024. All rights reserved.