题解 | #数组分组#

数组分组

http://www.nowcoder.com/practice/9af744a3517440508dbeb297020aca86

条件

  • 分成两组
  • 两组中各元素加起来的和相等
  • 所有5的倍数必须在其中一个组中
  • 所有3的倍数在另一个组中(不包括5的倍数)
  • 不是5的倍数也不是3的倍数能放在任意一组,可以将数组分为空数组

分析

  1. 既然提到了 5 的倍数和 3 的倍数不能放在一组,那首先要做的就是将数组筛选一遍:5 的倍数放在 A 组,3 的倍数放在 B 组,其余数字放在 C 组等待继续分配
  2. 需要 A 组和 B 组数字的和相等,可以转化为:C 组数字的经过任意的加法和减法计算,最终得到的值,等于 A 组数字之和和 B 组数字之和的差的绝对值

代码

def func2(c, n):
    if not c:
        if n == 0:
            return True
        else:
            return False
        
    c2 = c[:]
    i = c2.pop(0)
    if func2(c2, n+i) or func2(c2, n-i):
        return True
    else:
        return False
    

def func(s):
    a = []  # First group
    b = []  # Second group
    c = []  # Wait
    for i in s:
        if i % 5 == 0:
            a.append(i)
        elif i % 3 == 0:
            b.append(i)
        else:
            c.append(i)

    n = abs(sum(a) - sum(b))
    if sum(c) == n:
        return True
#     elif sum(c) < n:
#         return False
    else:
        return func2(c, n)


input()
s = input().split()
if func(int(_) for _ in s):
    print("true")
else:
    print("false")

# 5
# 1 0 2 3 -2
# true
全部评论

相关推荐

totoroyyw:千年老妖😂
投递华为等公司10个岗位
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
1
1
分享
牛客网
牛客企业服务