题目:输入两个链表,找出它们的第一个公共节点。
def first_common_node(head1,head2):
p1,p2 = head1,head2
cnt1,cnt2=0,0
while p1:
cnt1 +=1
p1=p1.next
while p2:
cnt2 +=1
p2=p2.next
p1, p2 = head1, head2
if cnt1>cnt2:
for i in range(cnt1-cnt2):
p1=p1.next
if cnt2>cnt1:
for i in range(cnt2-cnt1):
p2=p2.next
while p1.data!=p2.data:
p1=p1.next
p2=p2.next
return p1
注:要找出两个链表第一个公共节点,只要让两个链表在各自某个节点之后长度一致,然后一起往后遍历,当遇到两个节点相同时,即为第一个公共节点。所以首先算出两个链表的长度,然后让长的链表先走,使其剩余长度等于另一个链表,然后一起遍历即可。
第二种方法为用两个栈存储两个链表,然后两个栈做弹出操作,判断两个栈弹出的节点是否相同,最后一个相同的节点为两个链表第一个公共节点。