Pythonで指定された条件を満たす配列の要素を配置できるかどうかを確認します

numsという配列があるとします。条件に従うようにnumsの要素を再配置できるかどうかを確認する必要があります-

したがって、入力がnums =[8、-4、4、-8]のようである場合、出力は、i =0に対して[-4、-8、4、8]のように配列を配置するかのようにTrueになります。 nums [2 * 0 + 1] =2 *（-4）=-8 for i =1、nums [2 * 1 + 1] =2 * 4 =8

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

• freq：=numsの要素とその頻度を含むマップ
• 絶対値でソートされたnumの各項目について、
  を実行します。
  • freq [item]が0の場合、
    • 次の反復に進む
  • freq [2 * item]が0の場合、
    • Falseを返す
  • freq [item]：=freq [item]-1
  • freq [2 * item]：=freq [2 * item]-1
• Trueを返す

例

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

from collections import defaultdict
def solve(nums):
   freq = defaultdict(int)
   for item in nums:
      freq[item] += 1
   for item in sorted(nums, key = abs):
      if freq[item] == 0:
         continue
      if freq[2 * item] == 0:
         return False
      freq[item] -= 1
      freq[2 * item] -= 1
   return True
nums = [8, -4, 4, -8]
print(solve(nums))

入力

[8, -4, 4, -8]

出力

True