leetcode 160 相交链表

编写一个程序，找到两个单链表相交的起始节点。

例如，下面的两个链表：

A:          a1 → a2
                   ↘
                     c1 → c2 → c3
                   ↗            
B:     b1 → b2 → b3

在节点 c1 开始相交。

注意：

如果两个链表没有交点，返回 null.
在返回结果后，两个链表仍须保持原有的结构。
可假定整个链表结构中没有循环。
程序尽量满足 O(n) 时间复杂度，且仅用 O(1) 内存。

解法1：

首先分别遍历两个链表，得到分别对应的长度。然后求长度的差值，第二次遍历时把较长的那个链表向后移动这个差值的个数，找到的第一个相同的节点就是他们的第一个公共节点。

public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
    if(headA == null || headB == null) return null;
    int lenA = getLength(headA);
    int lenB = getLength(headB);
    if(lenA > lenB){
        for(int i = 0; i < lenA - lenB; i++){
            headA = headA.next;
        }
    }else{
        for(int i = 0; i < lenB - lenA; i++){
            headB = headB.next;
        }
    }
    while(headA != null && headB != null){
        if(headA == headB) break;
        headA = headA.next;
        headB = headB.next
    }
    return headA;
}
public int getLength(ListNode head){//得到链表的长度
    int count = 0;
    while(head != null){
        count++;
        head = head.next;
    }
    return count;
}

解法2：

但是我们可以用环的思想来做，我们让两条链表分别从各自的开头开始往后遍历，当其中一条遍历到末尾时，我们跳到另一个条链表的开头继续遍历。两个指针最终会相等，而且只有两种情况，一种情况是在交点处相遇，另一种情况是在各自的末尾的空节点处相等。为什么一定会相等呢，因为两个指针走过的路程相同，是两个链表的长度之和，所以一定会相等。

public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
    if(headA == null || headB == null) return null;
    ListNode a = headA, b = headB;
    while(a != b){
        a = (a != null) ? a.next : headB;
        b = (b != null) ? b.next : headA;
    }
    return a;
}

leetcode 160 相交链表

猜你喜欢