Flood Fill

功德箱我每年都投

于 2024-04-08 08:38:25 发布

阅读量880

点赞数 13

分类专栏： AcWing算法提高（Python代码）文章标签：算法 python 笔记

本文链接：https://blog.csdn.net/qq947467490/article/details/137430135

版权

AcWing算法提高（Python代码）专栏收录该内容

10 篇文章 0 订阅

订阅专栏

池塘计数

题目描述：

农夫约翰有一片 $\times M$ 的矩形土地。

最近，由于降雨的原因，部分土地被水淹没了。

现在用一个字符矩阵来表示他的土地。

每个单元格内，如果包含雨水，则用 $W$ 表示，如果不含雨水，则用 $.$ 表示。

现在，约翰想知道他的土地中形成了多少片池塘。

每组相连的积水单元格集合可以看作是一片池塘。

每个单元格视为与其上、下、左、右、左上、右上、左下、右下八个邻近单元格相连。

请你输出共有多少片池塘，即矩阵中共有多少片相连的 $W$ 块。

输入格式：

第一行包含两个整数 $N$ 和 $M$ 。

接下来 $N$ 行，每行包含 $M$ 个字符，字符为 ” $W$ ” 或 ” $.$ ”，用以表示矩形土地的积水状况，字符之间没有空格。

输出格式：

输出一个整数，表示池塘数目。

数据范围：

$\le N, M \le 1000$

输入样例：

10 12
W........WW.
.WWW.....WWW
....WW...WW.
.........WW.
.........W..
..W......W..
.W.W.....WW.
W.W.W.....W.
.W.W......W.
..W.......W.

输出样例：

代码实现

import sys
from collections import deque
input = sys.stdin.readline

n, m = map(int, input().strip().split())
grid = [input().strip() for _ in range(n)]

st = set()


def bfs(sx, sy):
    q = deque([(sx, sy)])
    st.add(sx * m + sy)
    while q:
        x, y = q.popleft()
        for tx in [x, x + 1, x - 1]:
            for ty in [y, y + 1, y - 1]:
                if tx == x and ty == y:
                    continue
                if 0 <= tx < n and 0 <= ty < m and grid[tx][ty] == 'W' and tx * m + ty not in st:
                    st.add(tx * m + ty)
                    q.append((tx, ty))


ans = 0
for i in range(n):
    for j in range(m):
        if grid[i][j] == 'W' and i * m + j not in st:
            bfs(i, j)
            ans += 1
print(ans)

城堡问题

题目描述：

     1   2   3   4   5   6   7  
   #############################
 1 #   |   #   |   #   |   |   #
   #####---#####---#---#####---#
 2 #   #   |   #   #   #   #   #
   #---#####---#####---#####---#
 3 #   |   |   #   #   #   #   #
   #---#########---#####---#---#
 4 #   #   |   |   |   |   #   #
   #############################
           (图 1)

   #  = Wall   
   |  = No wall
   -  = No wall

   方向：上北下南左西右东。

图 $1$ 是一个城堡的地形图。

请你编写一个程序，计算城堡一共有多少房间，最大的房间有多大。

城堡被分割成 $m * n$ 个方格区域，每个方格区域可以有 $0$ ~ $4$ 面墙。

注意：墙体厚度忽略不计。

输入格式：

第一行包含两个整数 $m$ 和 $n$ ，分别表示城堡南北方向的长度和东西方向的长度。

接下来 $m$ 行，每行包含 $n$ 个整数，每个整数都表示平面图对应位置的方块的墙的特征。

每个方块中墙的特征由数字 $P$ 来描述，我们用 $1$ 表示西墙， $2$ 表示北墙， $4$ 表示东墙， $8$ 表示南墙， $P$ 为该方块包含墙的数字之和。

例如，如果一个方块的 $P$ 为 $3$ ，则 $3 = 1 + 2$ ，该方块包含西墙和北墙。

城堡的内墙被计算两次，方块 $(1, 1)$ 的南墙同时也是方块 $(2, 1)$ 的北墙。

输入的数据保证城堡至少有两个房间。

输出格式：

共两行，第一行输出房间总数，第二行输出最大房间的面积（方块数）。

数据范围：

$1 \leq m, n \leq 50$ , $0 \leq P \leq 15$

输入样例：

4 7 
11 6 11 6 3 10 6 
7 9 6 13 5 15 5 
1 10 12 7 13 7 5 
13 11 10 8 10 12 13

输出样例：

5
9

代码实现

import sys
from collections import deque
input = sys.stdin.readline

n, m = map(int, input().strip().split())
grid = [list(map(int, input().strip().split())) for _ in range(n)]
dire = [(0, -1), (-1, 0), (0, 1), (1, 0)]
st = set()

def bfs(sx, sy):
    q = deque([(sx, sy)])
    area = 1
    st.add(sx * m + sy)
    while q:
        x, y = q.popleft()
        for i in range(4):
            if grid[x][y] >> i & 1:
                continue
            tx, ty = x + dire[i][0], y + dire[i][1]
            if 0 <= tx < n and 0 <= ty < m and tx * m + ty not in st:
                area += 1
                q.append((tx, ty))
                st.add(tx * m + ty)
    return area

cnt = max_area = 0
for i in range(n):
    for j in range(m):
        if i * m + j not in st:
            max_area = max(max_area, bfs(i, j))
            cnt += 1
print(cnt)
print(max_area)

山峰和山谷

问题描述：

$FG D$ 小朋友特别喜欢爬山，在爬山的时候他就在研究山峰和山谷。

为了能够对旅程有一个安排，他想知道山峰和山谷的数量。

给定一个地图，为 $FG D$ 想要旅行的区域，地图被分为 $n\times n$ 的网格，每个格子 $(i, j)$ 的高度 $w (i ， j)$ 是给定的。

若两个格子有公共顶点，那么它们就是相邻的格子，如与 $(i ， j)$ 相邻的格子有
$(i - 1 ， j - 1) ， (i - 1 ， j) ， (i - 1 ， j + 1) ， (i ， j - 1) ， (i ， j + 1) ， (i + 1 ， j - 1) ， (i + 1 ， j) ， (i + 1 ， j + 1)$ 。

我们定义一个格子的集合 $S$ 为山峰（山谷）当且仅当：

$S$ 的所有格子都有相同的高度。
$S$ 的所有格子都连通。
对于 $s$ 属于 $S$ ，与 $s$ 相邻的 $s'$ 不属于 $S$ ，都有 $w_s>w_{s′}$ （山峰），或者 $w_s<w_{s′}$ （山谷）。
如果周围不存在相邻区域，则同时将其视为山峰和山谷。

你的任务是，对于给定的地图，求出山峰和山谷的数量，如果所有格子都有相同的高度，那么整个地图即是山峰，又是山谷。

输入格式：

第一行包含一个正整数 $n$ ，表示地图的大小。接下来一个 $n \times n$ 的矩阵，表示地图上每个格子的高度 $w$ 。

输出格式：

共一行，包含两个整数，表示山峰和山谷的数量。

数据范围：

$1 ≤ n ≤ 1000，0 ≤ w ≤ 10 ^ 9$

输入样例1：

输出样例1：

2 1

输入样例2：

输出样例2：

3 3

样例解释

样例1：
在这里插入图片描述

样例2：
在这里插入图片描述

代码实现

import sys
from collections import deque
input = sys.stdin.readline

n = int(input().strip())
grid = [list(map(int, input().strip().split())) for _ in range(n)]
st = set()

def bfs(sx, sy):
    has_high = has_low = False
    val = grid[sx][sy]
    q = deque([(sx, sy)])
    st.add(sx * n + sy)
    while q:
        x, y = q.popleft()
        for dx in [-1, 0, 1]:
            for dy in [-1, 0, 1]:
                if dx == dy == 0:
                    continue
                tx, ty = x + dx, y + dy
                if 0 <= tx < n and 0 <= ty < n:
                    if grid[tx][ty] > val:
                        has_high = True
                    elif grid[tx][ty] < val:
                        has_low = True
                    elif grid[tx][ty] == val and tx * n + ty not in st:
                        st.add(tx * n + ty)
                        q.append((tx, ty))
    return has_high, has_low


cnt1 = cnt2 = 0
for i in range(n):
    for j in range(n):
        if i * n + j not in st:
            has_high, has_low = bfs(i, j)
            if not has_high:
                cnt1 += 1
            if not has_low:
                cnt2 += 1

print(cnt1, cnt2)