可能的二分法 -- 二分图判定【DFS、BFS分别实现】

886. 可能的二分法

python 复制代码
class PossibleBipartition:
    """
    可能的二分法
    「其实考察的就是二分图的判定」
    用dfs和bfs 两种方法分别实现
    https://leetcode.cn/problems/possible-bipartition/
    """
    def __init__(self):
        self.success = True
        self.color = []
        self.visited = []

    def dfs(self, n, dislikes):
        """
        DFS递归实现
        :param n: 
        :param dislikes:
        :return:
        """
        # 图节点编号为 1...n
        self.color = [False] * (n+1)
        self.visited = [False] * (n+1)
        graph = self.buildgraph(n, dislikes)

        # 因为图不一定是联通的,可能存在多个子图
        # 所以要把每个节点都作为起点进行一次遍历
        # 如果发现任何一个子图不是二分图,整幅图都不是二分图
        for v in range(1, n+1):
            if not self.visited[v]:
                self.dfs_traverse(graph, v)

        return self.success

    def buildgraph(self, n, dislikes):
        graph = [[] for _ in range(n+1)]

        for edge in dislikes:
            v = edge[1]
            w = edge[0]
            # 无向图相当于双向图
            graph[v].append(w)
            graph[w].append(v)

        return graph

    def dfs_traverse(self, graph, v):
        if not self.success:
            return

        self.visited[v] = True
        for w in graph[v]:
            if not self.visited[w]:
                self.color[w] = not self.color[v]
                self.dfs_traverse(graph, w)
            else:
                if self.color[v] == self.color[w]:
                    self.success = False
                    return

    def bfs(self, n, dislikes):
        """
        BFS实现,用队列替代递归调用
        :param n:
        :param dislikes:
        :return:
        """
        # 图节点编号为 1...n
        self.color = [False] * (n + 1)
        self.visited = [False] * (n + 1)
        graph = self.buildgraph(n, dislikes)

        # 因为图不一定是联通的,可能存在多个子图
        # 所以要把每个节点都作为起点进行一次遍历
        # 如果发现任何一个子图不是二分图,整幅图都不是二分图
        for v in range(1, n + 1):
            if not self.visited[v]:
                self.bfs_traverse(graph, v)

        return self.success

    def bfs_traverse(self, graph, start):
        # 节点队列
        queue = []
        self.visited[start] = True
        queue.append(start)

        while queue and self.success:
            v = queue.pop(0)
            # 从节点 v 向所有相邻节点扩散
            for w in graph[v]:
                if not self.visited[w]:
                    # 相邻节点w没有被访问过
                    # 那么应该给节点w涂上和节点v不同的颜⾊
                    self.color[w] = not self.color[v]
                    # 标记 w 节点,并放⼊队列
                    self.visited[w] = True
                    queue.append(w)
                else:
                    if self.color[v] == self.color[w]:
                        self.success = False
                        return
相关推荐
侯小啾3 小时前
【03】C语言 强制类型转换 与 进制转换
c语言·数据结构·算法
Xの哲學3 小时前
Linux NAPI 架构详解
linux·网络·算法·架构·边缘计算
京东零售技术7 小时前
扛起技术大梁的零售校招生们 | 1024技术人特别篇
算法
爱coding的橙子7 小时前
每日算法刷题Day78:10.23:leetcode 一般树7道题,用时1h30min
算法·leetcode·深度优先
Swift社区7 小时前
LeetCode 403 - 青蛙过河
算法·leetcode·职场和发展
地平线开发者7 小时前
三种 Badcase 精度验证方案详解与 hbm_infer 部署实录
算法·自动驾驶
papership7 小时前
【入门级-算法-5、数值处理算法:高精度的减法】
算法·1024程序员节
lingran__8 小时前
算法沉淀第十天(牛客2025秋季算法编程训练联赛2-基础组 和 奇怪的电梯)
c++·算法
DuHz8 小时前
基于MIMO FMCW雷达的二维角度分析多径抑制技术——论文阅读
论文阅读·物联网·算法·信息与通信·毫米波雷达
Dragon_D.8 小时前
排序算法大全——插入排序
算法·排序算法·c·学习方法