zl程序教程

您现在的位置是:首页 >  其它

当前栏目

红与黑(flood fill)

fill
2023-09-14 09:01:26 时间

文章目录

Question

有一间长方形的房子,地上铺了红色、黑色两种颜色的正方形瓷砖。

你站在其中一块黑色的瓷砖上,只能向相邻(上下左右四个方向)的黑色瓷砖移动。

请写一个程序,计算你总共能够到达多少块黑色的瓷砖。

输入格式
输入包括多个数据集合。

每个数据集合的第一行是两个整数 W 和 H,分别表示 x 方向和 y 方向瓷砖的数量。

在接下来的 H 行中,每行包括 W 个字符。每个字符表示一块瓷砖的颜色,规则如下

1)‘.’:黑色的瓷砖;
2)‘#’:红色的瓷砖;
3)‘@’:黑色的瓷砖,并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。

当在一行中读入的是两个零时,表示输入结束。

输出格式
对每个数据集合,分别输出一行,显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。

数据范围
1≤W,H≤20
输入样例:
6 9
…#.
…#





#@…#
.#…#.
0 0
输出样例:
45

Ideas

连通块问题:给一个连通块的任何一个点都能遍历到所有点

Code

N = 25
dx = [0,0,-1,1]
dy = [1,-1,0,0]

def bfs(start):
    q = [start]
    state[start[0]][start[1]] = 1 # 起点标记走过
    cnt = 1
    while q:
        t = q.pop(0)
        # 拓展点
        for i in range(4):
            x = t[0] + dx[i]
            y = t[1] + dy[i]
            
            if  0<=x<m and 0<=y<n and lis[x][y] =='.' and not state[x][y]:
                cnt += 1
                state[x][y] = 1 # 标记走过
                q.append((x,y))
    return cnt
while 1:
    lis = []
    state = [[0 for i in range(N)] for j in range(N)] # 判重数组 0代表没走过 1代表走过
    n,m = list(map(int,input().strip().split()))
    if n and  m:
        for i in range(m):
            lis.append(input().strip())
        
        for i in range(m):
            for j in range(n):
                if lis[i][j] == '@':
                    start = (i,j)
        
        print(bfs(start))
    else:
        break