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习题汇总

你可能感兴趣的:(数据结构,c/c++,LeetCode)