做题总结 160.链表相交

LeetCode：给你两个单链表的头节点 headA 和 headB ，请你找出并返回两个单链表相交的起始节点。如果两个链表没有交点，返回 null 。

我的思路

计算链表A、B的长度count1、count2。临时指针curA、curB要同时指向距离A、B链表尾端相同的位置。因此长链表的指针往后移动（count1-count2 或者 count2-count1）。两个指针同时移动，如果同时指向一个节点，该节点就是相交节点。

Java 实现：

public class Solution {
    public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
        ListNode dummyheadA = new ListNode(-1);
        dummyheadA.next = headA;
        ListNode dummyheadB = new ListNode(-2);
        dummyheadB.next = headB;

        ListNode curA = dummyheadA;
        ListNode curB = dummyheadB;
        int count1=0,count2=0;//count1没有初始化
        while(curA.next!=null) {
            count1++;
            curA = curA.next;
        }

        while(curB.next!=null) {
            count2++;
            curB = curB.next;
        }

        curA = dummyheadA;
        curB = dummyheadB;
        if(count1 >= count2) {//A长
            for(int i=0; i<(count1-count2); i++) {
                curA = curA.next;
            }
        } else {//B长
            for(int i=0; i<(count2-count1); i++) {
                curB = curB.next;
            }
        }

        //AB长度一样
        while(curA != null) {
            if(curA == curB) {
                return curA;
            } else {
                curA = curA.next;
                curB = curB.next;
            }
        }
        return null;
    }
}

代码改进

1 本题不涉及增删，可以不用头节点
2 第23行到31行，有更简便的实现方式。因为目的是让指向长链表的指针后移，所以可以规定curA这个变量指向长节点。

从代码随想录截取的代码片段。

// 让curA为最长链表的头，lenA为其长度
if (lenB > lenA) {
    //1. swap (lenA, lenB);
    int tmpLen = lenA;
    lenA = lenB;
    lenB = tmpLen;
    //2. swap (curA, curB);
    ListNode tmpNode = curA;
    curA = curB;
    curB = tmpNode;
}
// 求长度差
int gap = lenA - lenB;
// 让curA和curB在同一起点上（末尾位置对齐）
while (gap-- > 0) {
    curA = curA.next;
}