Pythonでソートされた順序で二乗要素リストを見つけるプログラム

numsと呼ばれる数値のリストがあり、要素が昇順で並べ替えられているとすると、要素を2乗して、並べ替えられた順序で結果を返す必要があります。

したがって、入力がnums =[-8、-3、0、5、6]のような場合、出力は[0、9、25、36、64]

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

• n：=numsのサイズ
• l：=0
• r：=n-1
• インデックス：=n-1
• res：=numsと同じサイズのリストで、0で埋めます
• インデックス>=0の場合、実行
  • if | nums [l] |> | nums [r] |、次に
    • res [index]：=nums [l] * nums [l]
    • l：=l + 1
  • それ以外の場合、
    • res [index]：=nums [r] * nums [r]
    • r：=r-1
  • index：=index-1
• return res

例

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

def solve(nums):
   n = len(nums)
   l = 0
   r = n - 1
   index = n - 1
   res = [0 for i in range(len(nums))]
   while index >= 0:
      if abs(nums[l]) > abs(nums[r]):
         res[index] = nums[l] * nums[l]
         l += 1
      else:
         res[index] = nums[r] * nums[r]
         r -= 1
      index -= 1

   return res

nums = [-8, -3, 0, 5, 6]
print(solve(nums))

入力

[-8, -3, 0, 5, 6]

出力

[0, 9, 25, 36, 64]