Pythonでのフェアキャンディスワップ

AとBが2人の友達だとします。彼らはさまざまなサイズのキャンディーバーを持っています。ここで、A [i]はAが所有するキャンディーのi番目のバーのサイズであり、B[j]はBが所有するキャンディーのj番目のバーのサイズです。

彼らは友達なので、交換後、AとBの両方が同じ合計量のキャンディーを持つように、それぞれ1つのキャンディーバーを交換したいと考えています。 （人が持っているキャンディーの総量は、持っているキャンディーバーのサイズの合計です。）整数配列を返す必要があります。ans[0]は、Aが交換する必要のあるキャンディーバーのサイズです。 ans [1]は、Bが交換する必要のあるキャンディーバーのサイズです。複数の回答がある場合は、そのうちの1つのみを返します。

たとえば、A =[1、2]およびB =[2、3]の場合、出力は[1、2]

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

• Aの合計とBの合計の差を取り、それを2で割って、整数部分をdiffにします
• Bをセットに変換する
• for i in A
  • i – Bでdiffの場合、[i、i –diff]を返します

例

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

class Solution(object):
   def fairCandySwap(self, A, B):
      diff = (sum(A) - sum(B))//2
      B=set(B)
      for i in A:
         if i- diff in B:
            return [i,i-diff]
ob1 = Solution()
print(ob1.fairCandySwap([1,2], [2,3]))

入力

[1,2]
[2,3]

出力

[1,2]