一、题目
题目描述:
小华和小为是很要好的朋友,他们约定周末一起吃饭。
通过手机交流,他们在地图上选择了多个聚餐地点(由于自然地形等原因,部分聚餐地点不可达),求小华和小为都能到达的聚餐地点有多少个?
二、输入输出
输入描述:
第一行输入m和n,m代表地图的长度,n代表地图的宽度。
第二行开始具体输入地图信息,地图信息包含:
0 为通畅的道路
1 为障碍物(且仅1为障碍物)
2 为小华或者小为,地图中必定有且仅有2个(非障碍物)
3 为被选中的聚餐地点(非障碍物)
输出描述:
可以被两方都到达的聚餐地点数量,行末无空格。
三、示例
示例1:
输入输出示例仅供调试,后台判题数据一般不包含示例
输入:
4 4
2 1 0 3
0 1 2 1
0 3 0 0
0 0 0 0
输出:
2
说明:
第一行输入地图的长宽为3和4。
第二行开始为具体的地图,其中:3代表小华和小明选择的聚餐地点;2代表小华或者小明(确保有2个);0代表可以通行的位置;1代表不可以通行的位置。
此时两者能都能到达的聚餐位置有2处。
示例2:输入输出示例仅供调试,后台判题数据一般不包含示例
输入:
4 4
2 1 2 3
0 1 0 0
0 1 0 0
0 1 0 0
输出:
0
说明:
第一行输入地图的长宽为4和4。
第二行开始为具体的地图,其中:3代表小华和小明选择的聚餐地点;2代表小华或者小明(确保有2个);0代表可以通行的位置;1代表不可以通行的位置。
由于图中小华和小为之间有个阻隔,此时,没有两人都能到达的聚餐地址,故而返回0。
备注:
地图的长宽为m和n,其中:
4 <= m <= 100
4 <= n <= 100
聚餐的地点数量为 k,则
1< k <= 100
四、要求
时间限制:C/C++ 1秒,其他语言 2秒
空间限制:C/C++262144K,其他语言524288K
五、解题思路
六、参考代码
# -*- coding: utf-8 -*-
'''
@File : 2023-B-欢乐的周末.py
@Time : 2023/12/29 20:24:49
@Author : mgc
@Version : 1.0
@Desc : None
'''# 并查集类
class UnionFind:def __init__(self, n=0):self.parent = [i for i in range(n+1)]def find(self, x):if x != self.parent[x]:self.parent[x] = self.find(self.parent[x])return self.parent[x]def union(self, x, y):root_x = self.find(x)root_y = self.find(y)if root_x != root_y:self.parent[root_y] = root_x# 获取输入
input_params = [int(x) for x in input().split()]
m = input_params[0]
n = input_params[1]
matrix = []
for _ in range(m):matrix.append([int(x) for x in input().split()])uf = UnionFind(m * n)
positions = []
dinner = []# 遍历地图
for i in range(m):for j in range(n):if matrix[i][j] == 1:continueelse:if matrix[i][j] == 2:positions.append(i * n + j)elif matrix[i][j] == 3:dinner.append(i * n + j)# 进行并查集合并操作if i + 1 < m and matrix[i+1][j] != 1:uf.union(i * n + j, (i+1) * n + j)if i - 1 >= 0 and matrix[i-1][j] != 1:uf.union(i * n + j, (i-1) * n + j)if j + 1 < n and matrix[i][j+1] != 1:uf.union(i * n + j, i * n + j + 1)if j - 1 >= 0 and matrix[i][j-1] != 1:uf.union(i * n + j, i * n + j - 1)# 统计能够被两方都到达的聚餐地点数量
target1 = uf.find(positions[0])
target2 = uf.find(positions[1])
result = 0
if target1 == target2:for k in range(len(dinner)):if uf.find(dinner[k]) == target1:result += 1print(result)