深度优先,广度优先搜索——代码简单易懂

深度优先,广度优先搜素

深度优先搜索是一种遍历方式,不局限在树或是图中,还可以用其解决很多层级关系的问题。

网上关于dfs介绍代码都很多了,但个人感觉不够直观,为了简洁性,使用最直观的python描述。

数据结构:

class Node:
    data = ""
    neighbors = []

    def __init__(self, data, neighbors=None) -> None:
        if neighbors is None:
            neighbors = []
        self.data = data
        self.neighbors = neighbors

DFS

递归遍历

def access(cur: Node):
    print(cur.data)
    
def search_re(cur: Node):
    visisted = set()

    def _search(cur: Node):
        nonlocal visisted
        access(cur)
        visisted.add(cur)
        for neighbor in cur.neighbors:
            if neighbor not in visisted:
                _search(neighbor)

    _search(cur)

非递归遍历

def search_it(cur: Node):
    visisted = set()
    stack = []
    stack.append(cur)
    while stack:
        cur = stack.pop()
        if cur not in visisted:
            access(cur)
            visisted.add(cur)
            for neighbor in cur.neighbors:
                if neighbor not in visisted:
                    stack.append(neighbor) 

BFS

def bfs(cur: Node):
    visisted = set()
    queue = deque()
    queue.append(cur)
    while queue:
        cur = queue.popleft()
        if cur not in visisted:
            access(cur)
            visisted.add(cur)
            for neighbor in cur.neighbors:
                if neighbor not in visisted:
                    queue.append(neighbor)

测试

    # 构建graph
    # a- b - d
    #  \ |  |
    #   c - e

    a, b, c, d, e = Node("a"), Node("b"), Node("c"), Node("d"), Node("e")
    a.neighbors.extend([b, c])
    b.neighbors.extend([a, c, d])
    c.neighbors.extend([a, b, e])
    d.neighbors.extend([b, e])
    e.neighbors.extend([c, d])

    # search_it(a)  # a c e d b
    # search_re(a)  # a b c e d 
    # bfs(a) # a b c d e

参考

http://web.cs.unlv.edu/larmore/Courses/CSC477/bfsDfs.pdf

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值