剑指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