Pythonでk日後の刑務所の独房の状態を見つけるプログラム

バイナリリスト（リスト内の1と0）と別の値kがあるとします。 numsの各値は、刑務所の独房の状態を表します。1は占有されたセルを示し、0は空のセルを示します。セルに2つの隣接するセルがあり、両方が占有されているか、両方が空である場合、そのセルは毎日占有されます。それ以外の場合は、空になります。したがって、k日後の刑務所の独房の状態を見つける必要があります。

したがって、入力がnums =[1、0、1、0、0、0、0、0] k =1のような場合、出力は[0、1、1、0、1、1、1になります。 0]、最初と最後のインデックスは2つのネイバーを持つことができないため、占有できないことがわかります。

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

• 関数next_day_state（）を定義します。これは細胞を取ります
• new_cells：=セルのコピー
• new_cells [0]：=0、new_cells [7]：=0
• 1〜6の範囲のjについては、
  • cells[j-1]がcells[j+ 1]と同じ場合、
    • new_cells [j]：=1
  • それ以外の場合、
    • new_cells [j]：=0
• return new_cells
• メインの方法から次の手順を実行します。
• 見た：=新しい地図
• フラグ：=False、i：=0
• i
• ns：=next_day_state（cells）
• nsが表示されない場合は、
  • 見たとおりにnsをマーク
• それ以外の場合、
  • フラグ：=True
  • ループから抜け出す
• セル：=ns
• i：=i + 1

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

例

import copy
class Solution:
   def next_day_state(self, cells):
      new_cells = copy.copy(cells)
      new_cells[0] = 0
      new_cells[7] = 0
      for j in range(1, 7):
         if cells[j - 1] == cells[j + 1]:
            new_cells[j] = 1
         else:
            new_cells[j] = 0
      return new_cells

   def solve(self, cells, N):
      seen = dict()
      flag, i = False, 0

      while i < N:
         ns = self.next_day_state(cells)
         if tuple(ns) not in seen:
            seen[tuple(ns)] = True
         else:
            flag = True
            break
         cells = ns
         i += 1

      if flag:
         N = N % len(seen)
         i = 0
         while i < N:
            ns = self.next_day_state(cells)
            i += 1
            cells = ns
      return cells

ob = Solution()
nums = [1, 0, 1, 0, 0, 0, 0, 0]
k = 1
print(ob.solve(nums, k))

入力

[4, 7, 2, 5], 6

出力

[0, 1, 1, 0, 1, 1, 1, 0]