LeetCode_160. 相交链表

题目

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

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode *getIntersectionNode(struct ListNode *headA, struct ListNode *headB) {
    //此处编写代码
}

例如，下面的两个链表：

A: 　　　　a1 → a2
　　　　　　　　　↘
　　　　　　　　　　c1 → c2 → c3
　　　　　　　　　↗
B: 　b1 → b2 → b3
在节点 c1 开始相交。

注意：

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

题解

思路: 有两种思路，第一种是先遍历A链表，将最后一个节点的next指向A的头结点，然后用链表判环的方式去找环的起点即两个链表的交叉点，最后要把环断开还原链表。链表判环见141. 环形链表。
第二种思路是先遍历两个链表，算出两链表长度的差值N，然后较长链表先走N步然后两链表同时向后遍历，遇上相同节点即为相交点。
note:注意链表为空的情况。最后两链表一定会交于NULL。

struct ListNode *getIntersectionNode(struct ListNode *headA, struct ListNode *headB) {
    if (__builtin_expect(!!(!headA || !headB), 0)) return NULL;
    int lena = 0, lenb = 0, len;
    struct ListNode *p = headA, *q = headB;
    while (p != NULL) {
        lena++;
        p = p->next;
    }
    while (q) {
        lenb++;
        q = q->next;
    }
    p = headA;
    q = headB;    
    if (lena > lenb) {
        len = lena - lenb;
        while (len--) {
            p = p->next;
        }    
    } else {
        len = lenb - lena;
        while (len--) {
            q = q->next;
        }
    }
    while (p != q) {
        p = p->next;
        q = q->next;
    }
    return p;
}

ps : 这题代码写的好丑

LeetCode习题汇总