Pythonでゼロを複製する

整数の固定長配列があるとすると、ゼロが出現するたびに複製し、残りの要素を右側にシフトする必要があります。

元の配列の長さを超える要素は書き込まれないことに注意してください。

したがって、配列が[1,0,2,3,0,4,5,0]のようであるとすると、変更後は[1,0,0,2,3,0,0,4]>

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

• arrを別の配列arr2にコピーし、iとjを0に設定します
• whilei
• arr2 [i]がゼロの場合、
  • arr [i]：=0
  • iを1増やします
  • i
• else arr [i]：=arr2 [j]
• iとjを1増やします

例

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

class Solution(object):
   def duplicateZeros(self, arr):
      arr2 = [i for i in arr]
      i=0
      j = 0
      while i < len(arr):
         if not arr2[j]:
            arr[i] = 0
            i+=1
            if i<len(arr):
               arr[i] = 0
         else:
            arr[i] = arr2[j]
         j+=1
         i+=1
      return arr
ob1 = Solution()
print(ob1.duplicateZeros([1,0,2,3,0,4,5,0]))

入力

[1,0,2,3,0,4,5,0]

出力

[1,0,0,2,3,0,0,4]