编写一个程序,找到两个单链表相交的起始节点。
例如,下面的两个链表:
A: a1 → a2 ↘ c1 → c2 → c3 ↗ B: b1 → b2 → b3
在节点 c1 开始相交。
注意:
- 如果两个链表没有交点,返回
null
. - 在返回结果后,两个链表仍须保持原有的结构。
- 可假定整个链表结构中没有循环。
- 程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。
首先遍历两链表得到其长度,其次根据长度差,从对应位置开始遍历。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode *getIntersectionNode(struct ListNode *headA, struct ListNode *headB) {
struct ListNode *a, *b, *c, *d;
a = headA; c = headA;
b = headB; d = headB;
int l1 = 0;
int l2 = 0;
char ca;
if(a == NULL && b == NULL) return NULL;
while(a != NULL){
a = a -> next;
l1++;
}
while(b != NULL){
b = b -> next;
l2++;
}
int n = l1 - l2;
if(n >= 0) ca = 'a';
else ca = 'b';
if(ca == 'a'){
while(n-- > 0) c = c -> next;
}
else{
while(n++ < 0) d = d -> next;
}
while(c != NULL && d != NULL){
if(c -> val != d -> val){
c = c -> next;
d = d -> next;
}
else
return c;
}
return NULL;
}