[Python]华为面试题，交换两个数组的元素使之总和的差值最小。

最新推荐文章于 2025-01-14 16:05:34 发布

最新推荐文章于 2025-01-14 16:05:34 发布 · 747 阅读

文章标签：

#Python #华为 #面试 #C #C++

本文介绍了一种使用Python的itertools库解决NPC问题的方法。通过运用permulations和product函数实现组合与排列，找到两个列表组合中元素之和最接近的配对。

这是个NPC问题，所以就用穷举发了，在这里给出了一个用python的itertools的解法，个人认为比较简洁。


import itertools

def funcProduct(a, b):
    for c in itertools.permutations(b):
        for d in itertools.product(*[list(itertools.permutations(x)) for x in zip(a, c)]):
            yield zip(*d)

a = [1,2,3,4]
b = [5,6,700,800]
print min(funcProduct(a,b), key=lambda x:abs(sum(x[0])-sum(x[1])))