蓝桥杯之深搜
图深搜
首先理解用图深搜
利用深搜遍历整个图结点
找出点与点之间的关系
maz={'A':['B','C','D'],
'B':['A'],
'C':['A','D'],
'D':['A','C']}
利用字典可以表示
点A和B,C,D点相连…
利用visit数组进行标记
visit={'A':False,'B':False,'C':False,'D':False}
如果这个结点没被访问 那么在visit字典里他是用False表示
一旦结点被访问
则把visit中的结点位置置为True
进行深搜
最开始访问的结点为start(一定没有被访问过)
访问之后设visit为True
进行相连的结点判断(与A相连的结点为’B,C,D)
所以访问maz[‘A’]=[‘B’,‘C’,‘D’] 进行遍历
判断这个结点是否被访问过
是-----进行下一个结点判断
否-----开始进行这个结点的深搜
def dfs(start):
visit[start]=True
for i in maz[start]:
if visit[i]==False:
print("访问结点{}".format(i))
dfs(i)
完整代码
maz={'A':['B','C','D'],
'B':['A'],
'C':['A','D'],
'D':['A','C']}
visit={'A':False,'B':False,'C':False,'D':False}
def dfs(start):
visit[start]=True
for i in maz[start]:
if visit[i]==False:
print("访问结点{}".format(i))
dfs(i)
start='A'
print("访问结点A")
if visit[start]!=True:#未被搜过
dfs(start)
草地寻人
题目
蒜头君和他的朋友周末相约去召唤师峡谷踏青。他们发现召唤师峡谷的地图是由一块一块格子组成的,有的格子上是草丛,有的是空地。草丛通过上下左右4个方向扩展其他草丛形成一片草地, 任何一片草地中的格子都是草丛,并且所有格子之间都能通过,上下左右连通。如果用‘#"代表草丛,.代表空地,下面的峡谷中有2片草地。
##..
..##
处在同一个草地的2个人可以相互看到,空地看不到草地里面的人。他们发现有一个朋友不见了,现在需要分头去找,每个人负责一片草地,蒜头君想知道他们至少需要多少人。
输入格式
第一行输入n,m表示峡谷大小。
接下来输入n行字符串表示峡谷的地形。
输入格式
输出至少需要多少人
理解: 就是判断有多少个相连的草丛,一样利用深搜,碰到草丛就搜,知道周围没草丛了,这时候一片草丛被搜完
需要多少人就是看有多少个草丛
是否需要回溯------不需要
点与点的关系
一个草丛可以走上下左右,如果为#就代表可以走,如果是.就走不了
利用maz数组进行存储点与点的关系
利用direct数组存储可以走的方向
diret=[[0,1],[0,-1],[1,0],[-1,0]]
利用visit数组进行标记
设置与地图maz相同大小的visit的数组
初始化为0
如果maz[x][y]=='#' and visit[x][y]==0 代表这个草丛没走过,可以走
完整代码
direct=[[1,0],[-1,0],[0,1],[0,-1]] #4个方向
maz=[]
cnt=0
n,m=map(int,input().split())
for i in range(n):
li=list(input())#子数列
maz.append(li)
#创建地图
visit=[[0 for i in range(m)]for j in range(n)] #判断是否走过 0是没走过
def dfs(x,y):
#如果可以深搜 代表已经是‘#’且0了
visit[x][y]=1
for i in direct:
x1=x+i[0]
y1=y+i[1]
if x1<n and y1<m and y1>=0 and x1>=0 and visit[x1][y1]==0 and maz[x1][y1]=='#':
dfs(x1,y1)
#return #这样只会检查当前的一个位置
return #检查4个位子
for i in range(n): #遍历地图所有的位置
for j in range(m):
if maz[i][j]=='#' and visit[i][j]==0:
dfs(i,j)
cnt=cnt+1
print(cnt)
生日蛋糕
与草地寻人相似
但草地寻人是看多少个区域
这个是找每个区域多大
注意区分
题目
这块蛋糕是由R x C的网格构成,每个网格上面都放有不同的水果。蒜头君把这些水果分为两类,一类是自己喜欢吃的水果,用’#"来表示;一类是自己不喜欢吃的水果,用’.'来表示。
蒜头君对切出的蛋糕有如下要求:
-切出的蛋糕连成一块(可以不为矩形,但必须在网格上连通)
-切出的蛋糕只包含自己喜欢吃的水果请问,蒜头君最大可以吃到多大的蛋糕?
输入格式
n m(行 列)
字符地图
输出格式
数字(最大蛋糕数)
是否需要回溯--------不需要
此题不用进行回溯,因为算的最大蛋糕数 深搜成一片后,记录这一片蛋糕是多大就行
如果是回溯的话 还会从这篇蛋糕的另一个部分进行深搜,结果与第一次一样 没有必要
完整代码
# -*- coding: utf-8 -*-
#2024-03-31 13:44
#2024-03-31 14:00
#要返回最大的cnt
n,m=map(int,input().split())
direct=[[1,0],[-1,0],[0,1],[0,-1]]
maz=[]
visit=[[0 for i in range(m)]for j in range(n)] #visit 都是为0
cnt=0
maxcnt=0
for i in range(n):
li=list(input())
maz.append(li)
def dfs(x,y):
global cnt
cnt += 1
visit[x][y]=1
for i in range(4):
x1=x+direct[i][0]
y1=y+direct[i][1]
if x1>=0 and y1>=0 and x1<m and y1<n and maz[x1][y1]=='#' and visit[x1][y1]==0:
dfs(x1,y1)
return cnt
for i in range(n):
for j in range(m):
if maz[i][j]=='#' and visit[i][j]==0:
cnt=0
maxcnt=max(dfs(i,j),maxcnt)
print(maxcnt)
案例:
5 5
##...
###..
#.##.
#....
.....
输出 9
迷宫地图
题目
有一个n*m的地图,地图由’.’(可同行的路)、‘#’(迷宫的墙)、‘s’(起点)、’e‘(终点),求能走到终点的路径条数
输入格式
n m(行 列)
字符地图
输出格式
数字(路径条数)
找点与点的关系
同样是4个方向
利用maz进行地图存储
是否需要回溯-------需要
注意:
如果四个方向都走完了,代表无路可走,此路走不通,那么要进行回溯,把当前已经置为visit[x] [y]=1 的数组回溯成visit【x】【y】=0 以便下一种顺序还能走
visit数组进行标记
同理
完整代码
#2024-03-31 11:35
# -*- coding: utf-8 -*-
direct=[[1,0],[-1,0],[0,1],[0,-1]]
n,m=map(int,input().split())
maz=[]
visit=[[0 for i in range(n)] for j in range(n)]
start=[0,0]
end=[0,0]
cnt=0
for i in range(n):
li=list(input())
if 's' in li:
start[0]=i
start[1]=li.index('s')
if 'e' in li:
end[0] = i
end[1] = li.index('e')
maz.append(li)
def dfs(x, y):
global cnt
visit[x][y] = 1
if maz[x][y] == 'e': #这时候终点的visit已经为1了,为了不影响其他的路径
cnt += 1 #要在后面恢复
visit[x][y]=0
return 1
for i in range(4):
x1 = x + direct[i][0]
y1 = y + direct[i][1]
if x1 < n and y1 < m and x1 >= 0 and y1 >= 0 and visit[x1][y1] == 0 and (maz[x1][y1]== '.' or maz[x1][y1]=='e'):
dfs(x1, y1)
visit[x][y] = 0
dfs(start[0],start[1])
print(cnt)
也可以提前判断,如果是e的话 不用标记为1
直接return
def dfs(x, y):
global cnt
if maz[x][y] == 'e':
# 找到之后,visit还是为0 以便下一次判断
cnt += 1
return
visit[x][y]=1 #其他的直接置为1
for i in range(4):
x1 = x + direct[i][0]
y1 = y + direct[i][1]
if x1 < n and y1 < m and x1 >= 0 and y1 >= 0 and visit[x1][y1] == 0 and (maz[x1][y1]== '.' or maz[x1][y1]=='e'):
dfs(x1, y1)
visit[x][y] = 0 #要在4个方向都找过之后 恢复现场
案例
5 5
s...#
#.#.#
#.#.#
#.#.#
e...#
#输出2
5 5
s...#
#.#.#
..#.#
..#.#
e...#
#输出7
直系后代
题目
统计每位祖先有多少直系后代
输入格式
输入n(表示家谱中的总人数)
读入n-1行,每行有2个整数x,y(表示x是y的父母)
输出格式
输出n行,每行有一个整数,表示第i个人有多少个直系后代
是否回溯--------需要
要算每一个人后代-------对每一个结点进行深搜
故当一个结点搜完了之后 要恢复孩子结点的visit 以便孩子结点进行深搜
完整代码
# -*- coding: utf-8 -*-
n=int(input())
maz={}
#地图构建完毕
for i in range(n-1):
pa,so=map(int,input().split())
if pa not in maz:
maz[pa]=[so]
else:
maz[pa].append(so)
for i in range(n):
if i+1 not in maz.keys():
maz[i+1]=[]
visit=[0 for i in range(n+1)] #从0-n 忽略0的位置
def dfs(parent):
global cnt
cnt=0
visit[parent]=1#看过了
for i in maz[parent]:
if visit[i]==0: #没被访问过可以进行深搜
cnt=cnt+1 #当前的孩子算一个
cnt=cnt+dfs(i) #再加上孩子的孩子
#搜完所有的孩子
#返回直系个数
visit[parent]=0
return cnt
for i in range(n):
print(i+1,dfs(i+1))
案例
5
1 2
1 3
2 4
2 5
#输出
1 4
2 2
3 0
4 0
5 0