Pythonの塗りつぶし操作を使用して色で塗りつぶすプログラム

文字列「r」、「g」、および「b」として色を含む2Dグリッドがあるとします。カラーターゲットを使用して、行r、列cで塗りつぶし操作を実行する必要があります。フラッドフィル操作では、grid [r、c]（上/右/下/左）に接続され、grid [r、c]と同じ色で、ターゲットと同じ色のすべての要素を置き換える必要があります。

したがって、入力が次のような場合

その場合、出力は次のようになります

grid [0,0]に接続されている赤いセルが緑（ "g"）に置き換えられるため。

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

• 表示される新しいセットを定義する
• oldcolor：=matrix [r、c]
• 関数dfs（）を定義します。これにはi、jが必要です
• iとjが行列にあり、（i、j）が表示されず、matrix [i、j]がoldcolorと同じである場合、
  • 見たもののadd（i、j）
  • matrix [i、j]：=target
  • dfs（i + 1、j）
  • dfs（i、j + 1）
  • dfs（i、j-1）
  • dfs（i-1、j
• メインの方法から、次の手順を実行します-
• dfs（r、c）
• リターンマトリックス

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

例

class Solution:
   def solve(self, matrix, r, c, target):
      def dfs(i, j):
         if (
            i >= 0
            and i < len(matrix)
            and j >= 0
            and j < len(matrix[0])
            and (i, j) not in seen
            and matrix[i][j] == oldcolor
         ):
            seen.add((i, j))
            matrix[i][j] = target
            dfs(i + 1, j)
            dfs(i, j + 1)
            dfs(i, j - 1)
            dfs(i - 1, j)
      seen = set()
      oldcolor = matrix[r][c]
      dfs(r, c)
      return matrix
ob = Solution()
matrix = [ ["r", "r", "r"], ["r", "g", "b"], ["g", "b", "b"] ]
r = 0
c = 0
target = "g"
print(ob.solve(matrix, r, c, target))

入力

matrix = [
["r", "r", "r"],
["r", "g", "b"],
["g", "b", "b"] ]
r = 0
c = 0
target = "g"

出力

[ ['g', 'g', 'g'], ['g', 'g', 'b'], ['g', 'b', 'b']]