面试算法:查找重合链表的首个相交节点

作者: 望月从良 | 来源:发表于2017-01-09 11:53 被阅读63次

更详细的讲解和代码调试演示过程,请参看视频
如何进入google,算法面试技能全面提升指南

给定两个单向链表,这两个链表有可能会有重叠,情况如下:

这里写图片描述

两个单向链表从节点5开始重合,要求给定一个空间复杂度为O(1)的算法,返回两个链表相交时的第一个节点。依据上图,也就是返回节点5.

首先我们需要做的是,确保给定的两个单向链表,他们是相交的。这个很好确定,只要从头遍历两个链表,如果他们的尾节点是一样的话,那么这两个链表就是相交的,问题是,如何尽快找到他们相交的第一个节点。

最笨的办法是,先找到尾节点,然后去掉尾节点,然后再次遍历查找新的尾节点,然后再去掉,直到两个链表没有共同的尾节点,那么最后去掉的共同尾巴节点,则是两个链表的首次相交节点。这种做法可行,但是算法复杂度是O(n^2)。有没有更好的办法呢?

假设第一个链表,从头结点到首次相交节点,所经历的距离用T1来表示,根据上图例子,T1 = 5, 也就是第一个队列从头结点0开始,需要经历节点1,2,3,4也就是总共5个节点后才能到达节点5.

我们用T3 表示队列2从头节点开始,到达首次相交节点的距离。根据上图,T3 = 3.

我们用T2表示两队列相交部分的节点数.依据上图T2 = 5.

由此队列1的长度为: T1 + T2 (1)
队列2的长度为:T3 + T2 (2)

如果我们能算出T3的数值,那么我们从队列2的头结点出发,经过T3-1步后,就能达到首次相交节点。我们如何计算T3的数值呢?

对T3的计算,需要一个小技巧.我们把队列2进行反转,得到下面情形:

这里写图片描述

如果此时我们从队列1的头结点开始进行遍历,那么从上头的节点0开始出发,会到队列2的头结点0结束。这样,在反转后,如果再次从头遍历队列1的话,得到的长度就是:

T1 + T3 + 1 (3).

根据上面三个公式,我们便可以计算出T3来。

(3) - (1) = T1 + T3 + 1 - T1 - T2 = T3 - T2 + 1
(3) - (1) + (2) = T3 - T2 + 1 + T3 + T2 = 2*T3 + 1

由此,我们可以反解出T3, 有了T3,我们便可以得到两队列首次相交节点了。

这个算法除了需要遍历两个队列外,还需要对其中一个队列进行反转,无论是遍历还是反转,其算法复杂度都是O(n), 因此总算法复杂度是O(n).

代码实现:


public class ListIntersetChecker {
    private Node listHead1;
    private Node listHead2;
    private int firstListLen = 0;  //t1 + t2
    private int secondListLen = 0; // t3 + t2
    private int lenAfterReverse = 0; // t1 + t3
    private ListReverse listReverse = null;
    
    public ListIntersetChecker(Node listHead1, Node listHead2) {
        this.listHead1 = listHead1;
        this.listHead2 = listHead2;
        
    }
    
    public Node getFirstIntersetNode() {
        if (isTwoListInterset() == false) {
            return null;
        }
        
        listReverse = new ListReverse(listHead2);
        
        Node reverseHead = listReverse.getReverseList();
        lenAfterReverse = getListLen(listHead1);
        
        int t3 = ((lenAfterReverse - firstListLen) + secondListLen - 1) / 2;
        int steps = secondListLen - t3 - 1;
        while (steps > 0) {
            reverseHead = reverseHead.next;
            steps--;
        }
        
        return reverseHead;
    }
    
    private int getListLen(Node head) {
        int len = 0;
        while (head != null) {
            head = head.next;
            len++;
        }
        
        return len;
    }
    
    private boolean isTwoListInterset() {
        Node head1 = listHead1;
        Node head2 = listHead2;
        
        while (head1.next != null || head2.next != null) {
            if (head1.next != null) {
                head1 = head1.next;
                firstListLen++; 
            }
            
            if (head2.next != null) {
                head2 = head2.next;
                secondListLen++;
            }
            
        }
        
        firstListLen++;
        secondListLen++;
        
        return head1 == head2;
    }
    
    
}

ListIntersetCheck.java 用于实现上面描述的算法。 getFirstIntersetNode返回两重叠队列首次相交节点。isTwoListInterset 用于判断两队列是否相交。在遍历两队列时,统计两队列的长度,也就是获得 T1 + T2 以及 T3 + T2的值。

然后把队列2进行反转,反转后,再从队列1的头节点进行遍历,得到的lenAfterReverse就是 T1 + T3 + 1.

int t3 = ((lenAfterReverse - firstListLen) + secondListLen - 1) / 2;
上面语句则根据前面的推导计算出T3.

由于队列2已经反转了,所以不能从队列2的头结点去遍历,只能从队列2的尾节点开始遍历,如果头结点开始遍历需要T3步的话,那么从尾节点遍历,则需要steps = secondListLen - (T3 + 1) 步。

由此,代码从队列2反转后的头结点开始,经过steps个节点后抵达两队列首次相交时的节点。

再看看主入口代码:


public class LinkList {
    public static void main(String[] args) {
        ListUtility util1 = new ListUtility();
        ListUtility util2 = new ListUtility();
        
        Node list1 = util1.createList(10);
        Node list2 = util2.createList(3);
        
        Node node = util1.getNodeByIdx(5);
        Node tail = util2.getNodeByIdx(2);
        tail.next = node;
        
        ListIntersetChecker intersetChecker = new ListIntersetChecker(list1, list2);
        Node interset = intersetChecker.getFirstIntersetNode();
        System.out.println("The first interset node is : " + interset.val);
    }
}

程序启动时,先构造两个队列,队列1节点从0到9,队列2从0到2,然后把队列2的尾节点的next指向队列1的编号为5的节点,于是就构造了我们例子图中的两个相交队列,然后再利用ListIntersetChecker获得两重合队列的首个相交节点。

最后程序运行结果为:
The first interset node is : 5
结果跟我们理论推导一致,也就是说,我们的说法实现是正确的。更详细的代码讲解和推导调试过程,请参看视频。

更多技术信息,包括操作系统,编译器,面试算法,机器学习,人工智能,请关照我的公众号:


这里写图片描述

相关文章

  • 面试算法:查找重合链表的首个相交节点

    更详细的讲解和代码调试演示过程,请参看视频如何进入google,算法面试技能全面提升指南 给定两个单向链表,这两个...

  • 大厂面试系列(七):数据结构与算法等

    数据结构和算法 链表 链表,常见的面试题有写一个链表中删除一个节点的算法、单链表倒转、两个链表找相交的部分,这个一...

  • 1.数据结构-链表问题

    链表相关问题 删除节点 链表去重 有环链表 反转链表 链表排序 链表相交 其他问题 面试题 02.03. 删除中间...

  • leetcode的题目160

    160. 相交链表 编写一个程序,找到两个单链表相交的起始节点。 例如,下面的两个链表: 在节点 c1 开始相交。...

  • 2019-01-25 Day 20

    1.#### 相交链表编写一个程序,找到两个单链表相交的起始节点。 如下面的两个链表: 在节点 c1 开始相交。 ...

  • 160. 相交链表

    160. 相交链表 编写一个程序,找到两个单链表相交的起始节点。 如下面的两个链表: 在节点 c1 开始相交。 示...

  • LeetCode刷题分类之双指针160. 相交链表

    160. 相交链表 题目 编写一个程序,找到两个单链表相交的起始节点。 如下面的两个链表: 在节点 c1 开始相交...

  • 2020-04-21 阿里电话面试试题

    1、算法:怎么判断俩个链表是否有相交节点(例如Y 形状)?https://leetcode-cn.com/prob...

  • 【LeetCode】链表专题

    160.相交链表 返回两个链表相交的节点 19.删除链表的倒数第N个节点 返回链表的头结点。 21.合并两个有序链表

  • 求两个链表的交点

    已知链表A的头节点指针headA,链表B的头节点指针headB,两个链表相交,求两链表交点对应的节点。[](Lee...

网友评论

    本文标题:面试算法:查找重合链表的首个相交节点

    本文链接:https://www.haomeiwen.com/subject/cxthbttx.html