52-两个链表的第一个公共节点
题目:输入两个链表,找出它们的第一个公共节点。
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
注:要找出两个链表第一个公共节点,只要让两个链表在各自某个节点之后长度一致,然后一起往后遍历,当遇到两个节点相同时,即为第一个公共节点。所以首先算出两个链表的长度,然后让长的链表先走,使其剩余长度等于另一个链表,然后一起遍历即可。
第二种方法为用两个栈存储两个链表,然后两个栈做弹出操作,判断两个栈弹出的节点是否相同,最后一个相同的节点为两个链表第一个公共节点。