Pythonで最初のセットの要素が2番目のセットよりも小さいセットを等しい合計セットに分割するプログラム

numsと呼ばれる数値のリストがあるとすると、次のようにリストを2つのグループAとBに分割できるかどうかを確認する必要があります。Aの合計とBの合計が同じである。ここで、Aのすべての数値は、Bのすべての数値よりも厳密に小さくなっています。

したがって、入力がnums =[3、4、5、12]のような場合、A=[3,4,5]とB=[12]があり、両方の合計が12であるため、出力はTrueになります。 。

これを解決するには、次の手順に従います-

• リスト番号を並べ替える

• total：=numsのすべての要素の合計

• s：=0、i：=0

• 私は

  • n：=nums [i]

  • i

    • s：=s + nums [i]

    • i：=i + 1

  • sがtotal− sと同じ場合、

    • Trueを返す

• Falseを返す

理解を深めるために、次の実装を見てみましょう-

例

class Solution:
   def solve(self, nums):
      nums.sort()
      total = sum(nums)
      s = 0
      i = 0
      while i < len(nums):
         n = nums[i]
         while i < len(nums) and nums[i] == n:
            s += nums[i]
            i += 1
         if s == total - s:
            return True
      return False
ob = Solution()
nums = [3, 4, 5, 12]
print(ob.solve(nums))

入力

[3, 4, 5, 12]

出力

True