Leetcode-链表相交

54 阅读1分钟

如果两个链表相交,那么相交点之后的长度是相同的 判断两个链表是否相交,可以使用哈希集合存储链表节点。

首先遍历链表 headA,并将链表 headA中的每个节点加入哈希集合中。然后遍历链表 headB,对于遍历到的每个节点,判断该节点是否在哈希集合中:

如果当前节点不在哈希集合中,则继续遍历下一个节点;

如果当前节点在哈希集合中,则后面的节点都在哈希集合中,即从当前节点开始的所有节点都在两个链表的相交部分,因此在链表 headB 中遍历到的第一个在哈希集合中的节点就是两个链表相交的节点,返回该节点。

如果链表 headB 中的所有节点都不在哈希集合中,则两个链表不相交,返回 null

class Solution:
    def getIntersectionNode(self, headA, headB):
        # 创建一个空字典用于存储已访问的节点
        visited = {}

        # 第一次遍历链表 headA
        temp = headA
        while temp is not None:
            visited[temp] = True
            temp = temp.next

        # 第二次遍历链表 headB,查找是否有相交节点
        temp = headB
        while temp is not None:
            if temp in visited:
                # 如果找到相交节点,立即返回
                return temp
            temp = temp.next

        # 如果没有相交节点,返回 None
        return None

关键就是两次遍历链表 visited[temp] = True,这个无所谓的,只要把tmp弄进去就可以了