【面试题】实现文件夹中文件的遍历输出文章中,我们用递归和深度遍历的方式实现了循环遍历输入文件夹中的文件。有没有基于广度优先遍历文件的方式呢,其实是有的。我们看下具体是怎么实现的。
广度优先遍历,指的是从图的一个未遍历的节点出发,先遍历这个节点的相邻节点,再依次遍历每个相邻节点的相邻节点。
根据广度优先的概念,去实现广度优先版本的文件遍历。
对应的思路:
1.创建一个队列
2.队列增加文件路径
3.当队列不为空,获取队列
4.遍历某个路径,判断是文件输出,是文件夹加入队列
5.直到队列为空,程序终止运行。
上面的思路出来了,我们基于思路去实现对应的代码。
import os, collections
# 广度遍历目录
def Get_All_Dir_Scope(path:str):
#创建队列
queue = collections.deque()
# 进队
queue.append(path)
print("queue =", queue)
while len(queue) != 0:
# 出队数据
File_Path = queue.popleft()
# print(FilePath)
# 找出所有的文件
num = 0
file_num = 0
FileNameList = os.listdir(File_Path)
for fileName in FileNameList:
fileAbsPath = os.path.join(File_Path, fileName)
if os.path.isfile(fileAbsPath):
print("是文件", fileAbsPath)
num += 1
else:
file_num += 1
queue.append(fileAbsPath)
print('当前文件数量:%s' % num, '当前文件夹数量%s' % file_num, '路径是:%s' % (fileAbsPath))
我们去传递一个路径,遍历里面文件
path = '.'
Get_All_Dir_Scope(path)
最后的打印结果
可以看到上面的结果,和我们想要的输出是一致的。
基于广度遍历的方式就实现完毕。其实很简单。