题目
编写一个程序,找到两个单链表相交的起始节点。
/**
* 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 : 这题代码写的好丑