52-两个链表的第一个公共节点

Posted 2020-11-13 尘世中一个迷途小书童

题目：输入两个链表，找出它们的第一个公共节点。

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

　　注：要找出两个链表第一个公共节点，只要让两个链表在各自某个节点之后长度一致，然后一起往后遍历，当遇到两个节点相同时，即为第一个公共节点。所以首先算出两个链表的长度，然后让长的链表先走，使其剩余长度等于另一个链表，然后一起遍历即可。

第二种方法为用两个栈存储两个链表，然后两个栈做弹出操作，判断两个栈弹出的节点是否相同，最后一个相同的节点为两个链表第一个公共节点。