两个链表的第一个公共节点
输入两个链表,找出它们的第一个公共结点。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)
思路:首先我们要知道什么是公共节点,两个链表从某一节点开始,他们的next都指向同一个节点。
但由于是单向链表的节点,每个节点只有一个next,因此从第一个公共节点开始,之后他们的所有节点都是重合的,不可能再出现分叉。所以可以先遍历两个链表得到他们的长度,就能知道哪个链表比较长,以及长的链表比短的链表多几个结点。在第二次遍历的时候,在较长的链表上先走若干步**,接着同时在两个链表上遍历,找到的第一个相同的结点就是他们的第一个公共结点。**时间复杂度为O(m+n),而暴力破解法的时间复杂度为O(mn).
# -*- coding:utf-8 -*-
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def FindFirstCommonNode(self, pHead1, pHead2):
# write code here
# 如果两个链表一样长
p1 = pHead1
p2 = pHead2
lenght1 = lenght2 = 0
# 求出链表的长度
while p1:
lenght1 += 1
p1 = p1.next
while p2:
lenght2 += 1
p2 = p2.next
# 把phead1定义为最长的
if lenght1 < lenght2:
pHead1, pHead2 = pHead2, pHead1
distance = abs(lenght1-lenght2)
# 让长的先走k步
for i in range(distance):
pHead1 = pHead1.next
#如果不相等则循环,直至找到一样的节点
while pHead1 != pHead2:
pHead1 = pHead1.next
pHead2 = pHead2.next
return pHead1
思路二:定义两个指针, 第一轮让两个到达末尾的节点指向另一个链表的头部, 最后如果相遇则为交点(在第一轮移动中恰好抹除了长度差)
两个指针等于移动了相同的距离, 有交点就返回, 无交点就是各走了两条指针的长度
# -*- coding:utf-8 -*-
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def FindFirstCommonNode(self, pHead1, pHead2):
# write code here
a = pHead1
b = pHead2
while a != b:
a = a.next if a else pHead2
b = b.next if b else pHead1
return a