其他分享
首页 > 其他分享> > 剑指offer——两个链表的第一个公共节点

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

作者:互联网

题目描述
输入两个链表,找出它们的第一个公共结点。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)

解答如下:
环境: python 2.7.3

1.本题设置两个指针,p1指向pHead1,p2指向pHead2,然后让它们同时像后移动,如下图,当p1指向None时,开始向后移动p2,这样就可以统计两指针间距,其实也就是两链表的长度差,如下图k=2。(如果两者同时指向None,那么它们间距为0,也就是两链表一样长,那么就可以直接进入下一步,不用先移动某个指针了。)
2.再让两指针同时指向表头,这时让长的链表的临时指针p2先后移k步,这样两临时指针到表尾的距离相同,再同时向后移动两指针,它们相遇的地方就是两链表公共节点。

在这里插入图片描述
算法我没有做精简优化,读者可以自行优化,基本思想是对的。

class Solution:
    def FindFirstCommonNode(self, pHead1, pHead2):
        # write code here
        if pHead1 == None or pHead2 == None:
            return None
        
        p1 = pHead1
        p2 = pHead2
        while p1 and p2:
            p1 = p1.next
            p2 = p2.next
            
        count = 0
        if p1 == None and p2 != None:
            while p2 != None:
                p2 = p2.next
                count += 1
            p1 = pHead1
            p2 = pHead2
            for i in range(count):
                p2 = p2.next
            
        if p1 != None and p2 == None:
            while p1 != None:
                p1 = p1.next
                count += 1
            p1 = pHead1
            p2 = pHead2
            for i in range(count):
                p1 = p1.next
        
        if p1 == None and p2 == None:
            return pHead1
            
        while p1 != p2:
            p1 = p1.next
            p2 = p2.next
        
        return p1

标签:p2,None,p1,offer,next,链表,pHead1,节点
来源: https://blog.csdn.net/QQ32162591/article/details/104880657