1062. 洪水填充(经典)

class Solution:
    """
    @param image: a 2-D array
    @param sr: an integer
    @param sc: an integer
    @param newColor: an integer
    @return: the modified image
    """
    ‘‘‘ 
    大致思路:
    1.给出一个方法，可以求得当前像素相同颜色的像素，4方向相连，返回未添加的像素点列表。
    2.初始化res,result = [],[]，循环image，首先append到res初始值,image[sr][sc],然后调用方法，传入参数，最终得到所有4向相连的列表组合。
    3.result作为一个循环添值减值的列表，所以这个作为循环判断条件，如果pop完所有的值，则break
    ‘‘‘

    def floodFill(self, image, sr, sc, newColor):
        res = []
        #给定初始值
        origin_color = image[sr][sc]
        res,result = [[sr,sc]],[[sr,sc]]


        while  result != []:
            pop_column = result.pop(0)
            image[pop_column[0]][pop_column[1]] = newColor
            #pop_column格式:[x,y]
            r_list = self.getaroundcolumn(image,pop_column,origin_color,res)
            res.extend(r_list)
            result.extend(r_list)

        return image



    ##得到当前像素四向相连的像素列表集合
    def getaroundcolumn(self,image,num,origin_color,res):
        around_list = []
        x,y = num[0],num[1]
        ##不管是哪个元素，进来都判断4个点，即上下左右四个像素
        ‘‘‘
        符合条件，则append到self.res里面
        1.x，y范围在[0,len(image)]和[0,len(image[0])]
        2.当前像素不存在self.res里面
        ‘‘‘

        #左方向:i不变，y-1
        if y-1 >= 0:
            if [x,y-1] not in res and image[x][y-1] == origin_color:
                around_list.append([x,y-1])
        
        #右方向
        if y + 1 < len(image[0]):
            if [x,y+1] not in res and image[x][y+1] == origin_color:
                around_list.append([x,y+1])

        #上方向
        if x-1 >= 0:
            if [x-1,y] not in res and image[x-1][y] == origin_color:
                around_list.append([x-1,y])
        
        #下方向
        if x+1 < len(image):
            if [x+1,y] not in res and image[x+1][y] == origin_color:
                around_list.append([x+1,y])
        return around_list