剑指offer 两个链表的第一个公共节点

题目描述:
输入两个无环的单向链表,找出它们的第一个公共结点,如果没有公共节点则返回空。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)

例如,输入{1,2,3},{4,5},{6,7}时,两个无环的单向链表的结构如下图所示:
剑指offer 两个链表的第一个公共节点_第1张图片
可以看到它们的第一个公共结点的结点值为6,所以返回结点值为6的结点。

输入描述:
输入分为是3段,第一段是第一个链表的非公共部分,第二段是第二个链表的非公共部分,第三段是第一个链表和第二个链表的公共部分。 后台会将这3个参数组装为两个链表,并将这两个链表对应的头节点传入到函数FindFirstCommonNode里面,用户得到的输入只有pHead1和pHead2。
返回值描述:
返回传入的pHead1和pHead2的第一个公共结点,后台会打印以该节点为头节点的链表。

解题思路:
使用两个指针 a,b 分别指向两个链表 pHead1,pHead2的头结点,然后同时分别逐结点遍历,当 a 到达链表 pHead1的末尾时,重新定位到链表 pHead2的头结点;当 b 到达链表 pHead2 的末尾时,重新定位到链表 pHead1的头结点。当双指针相遇时,所指向的结点就是第一个公共结点。

Python代码实现如下:

class Solution:
    def FindFirstCommonNode(self , pHead1 , pHead2 ):
        # write code here
        a=pHead1
        b=pHead2
        while a!=b:
            # a遍历完pHead1再遍历pHead2
            a=a.next if a else pHead2
            # b遍历完pHead2再遍历pHead1
            b=b.next if b else pHead1
        return a

你可能感兴趣的:(Python,剑指offer,链表,数据结构,算法)