检查Python中是否可以将一个包含1和2的数组分成两个和相等的子数组

假设我们有一个数组nums，其中只存储1和2。我们需要检查该数组是否可以分成两个不同的部分，使得每个部分的元素之和相同。

因此，如果输入类似于nums = [1, 1, 2, 2, 2]，则输出将为True，因为我们可以将此数组分成[1, 1, 2]和[2, 2]，每个部分的和都是4。

为了解决这个问题，我们将遵循以下步骤：

• total := 0，one_count := 0
• total := nums中所有元素的和
• one_count := nums中1的数量
• 如果total是偶数，则
  • 返回False
• 如果(total / 2)的整数部分是偶数，则
  • 返回True
• 如果one_count > 0，则
  • 返回True
• 否则，
  • 返回False

让我们看看下面的实现来更好地理解：

示例

 实时演示

def solve(nums):
   total = 0
   one_count = 0
   total = sum(nums)
   one_count = nums.count(1)
   if total % 2:
      return False
   if (total // 2) % 2 == 0:
      return True
   if one_count > 0:
      return True
   else:
      return False
nums = [1, 1, 2, 2, 2]
print(solve(nums))

输入

[1, 1, 2, 2, 2]

输出

True

Arnab Chakraborty

更新于: 2020-12-30

131 次查看

开启你的 职业生涯

通过完成课程获得认证

开始学习