其他分享
首页 > 其他分享> > 剑指Offer【36】-- 两个链表的第一个公共节点

剑指Offer【36】-- 两个链表的第一个公共节点

作者:互联网

【刷题笔记】
Github仓库地址:https://github.com/Damaer/codeSolution
笔记地址:https://damaer.github.io/codeSolution/

题目描述

输入两个链表,找出它们的第一个公共结点。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)

思路以及解答

HashSet是否包含

第一种做法,直接依赖于HashSet,遍历第一个链表的时候,将所有的节点,添加到hashset中,遍历第二个链表的时候直接判断是否包含即可,属于空间换时间的做法。

    public ListNode FindFirstCommonNode(ListNode pHead1, ListNode pHead2) {
        //创建集合set
        Set<ListNode> set = new HashSet<>();

        while (pHead1 != null) {
            set.add(pHead1);
            pHead1 = pHead1.next;
        }

        while (pHead2 != null) {
            if (set.contains(pHead2))
                return pHead2;
            pHead2 = pHead2.next;
        }
        return null;
    }

计算个数提前移动差值

譬如现在有一个链表1->2->3->6->7,另外一个链表4->5->6->7,明显可以看出第一个公共节点是 6 。

最直接的方法,每一个链表都遍历一次,计算链表中的个数,比如1->2->3->6->7个数为5,4->5->6->7个数为4,两者相差1(设为k)个。

我们可以使用两个指针,分别指向链表的头部。然后让第一个链表的指针先走k=1步。

这样就相当于指针后面的两个链表等长了。

就可以开始比较,如果不相等,则两个指针都往后移动即可,知道节点为null。

代码如下:

/*
public class ListNode {
    int val;
    ListNode next = null;

    ListNode(int val) {
        this.val = val;
    }
}*/public class Solution {
    public ListNode FindFirstCommonNode(ListNode pHead1, ListNode pHead2) {
        // 只要有一个为空,就不存在共同节点
        if (pHead1 == null || pHead2 == null) {
            return null;
        }

        // 计算链表1中的节点个数
        int numOfListNode1 = 0;
        ListNode head1 = pHead1;
        while (head1 != null) {
            numOfListNode1++;
            head1 = head1.next;
        }

        // 计算链表2中节点个数
        int numOfListNode2 = 0;
        ListNode head2 = pHead1;
        while (head2 != null) {
            numOfListNode2++;
            head2 = head2.next;
        }

        // 比较两个链表的长度
        int step = numOfListNode1 - numOfListNode2;
        if (step > 0) {
            // 链表1更长,链表1移动
            while (step != 0) {
                pHead1 = pHead1.next;
                step--;
            }
        } else {
            // 链表2更长,链表2移动
            while (step != 0) {
                pHead2 = pHead2.next;
                step++;
            }
        }

        // 循环遍历后面的节点,相等则返回
        while (pHead1 != null && pHead2 != null) {
            if (pHead1 == pHead2) {
                return pHead1;
            } else {
                pHead1 = pHead1.next;
                pHead2 = pHead2.next;
            }
        }
        return null;
    }}

但是上面的做法,如果公共节点在最后一个,假设一个链表长度为n,一个为m,那么计算个数就要全部遍历,需要n+m。两个链表都移动,到最后一个节点的时候才相等,也是n +m,也就是O(2*(n+m))。

拼接链表

有没有更加好用的做法呢?肯定有,我们来看:
两个链表分别是:

如果我在第一个链表后面拼接上第二个链表,第二个链表后面拼接上第一个链表,就会变成下面的样子:

发现了一个规律,也就是拼接之后的链表,是等长度的,第一个和第二个链表都从第一个开始比较,只要相等,就说明是第一个公共节点。也就是上面被圈起来的6节点。

代码如下:

public class Solution {
    public ListNode FindFirstCommonNode(ListNode pHead1, ListNode pHead2) {
        // 只要有一个为空,就不存在共同节点
        if (pHead1 == null || pHead2 == null) {
            return null;
        }
        ListNode head1 = pHead1;
        ListNode head2 = pHead2;
        while (head1 !=head2) {
              // 如果下一个节点为空,则切换到另一个链表的头节点,否则下一个节点
              head1 = (head1 == null) ? pHead2 : head1.next;
              head2 = (head2 == null) ? pHead1 : head2.next;
        }
        return head1;
    }}

【作者简介】
秦怀,公众号【秦怀杂货店】作者,技术之路不在一时,山高水长,纵使缓慢,驰而不息。个人写作方向:Java源码解析,JDBC,Mybatis,Spring,redis,分布式,剑指Offer,LeetCode等,认真写好每一篇文章,不喜欢标题党,不喜欢花里胡哨,大多写系列文章,不能保证我写的都完全正确,但是我保证所写的均经过实践或者查找资料。遗漏或者错误之处,还望指正。

2020年我写了什么?

开源编程笔记

平日时间宝贵,只能使用晚上以及周末时间学习写作,关注我,我们一起成长吧~

标签:ListNode,Offer,--,链表,pHead1,pHead2,null,节点
来源: https://blog.51cto.com/u_13604316/2729777