编写一个程序,找到两个单链表相交的起始节点。
如下面的两个链表:
在节点 c1 开始相交。
输出节点C1
注意:
如果两个链表没有交点,返回 null.
在返回结果后,两个链表仍须保持原有的结构。
可假定整个链表结构中没有循环。
程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。
解题思路:
从等长位置,同时遍历,如果有交点,一定会有 pA==pB
解法1:
求出2个链表长度,长的链表 表头先向后偏移,确保和短的链表 表头处于一个位置。在一起遍历。代码如下:
struct node {
int value;
node* next;
};
node* get_intersection_node(node* headA, node* headB) {
auto temp = headA;
int lenghta = 0;
for (; temp != nullptr;) {
++lenghta;
temp = temp->next;
}
temp = headB;
int lenghtb = 0;
for (; temp != nullptr;) {
++lenghtb;
temp = temp->next;
}
if (lenghta > lenghtb) { // 链表A比B长 ,偏移 链表A的head, 使对比长度一样
auto deata = lenghta - lenghtb;
for (int i = 0; i < deata; ++i) {
headA = headA->next;
}
}
else if(lenghta < lenghtb) {// 链表B比a长 ,偏移 链表B的head, 使对比长度一样
auto deata = lenghtb - lenghta;
for (int i = 0; i < deata; ++i) {
headB = headB->next;
}
}
for (; headA != nullptr && headB != nullptr;) {
if (headA == headB) { //地址相同,则有交点
return headA;
}
headA = headA->next;
headB = headB->next;
}
return nullptr;
}
解法2:
2个链表长度不一样,短的遍历完接后遍历长的,长的遍历完接后遍历短的。
链表A后接链表B:a1->a2->c1->c2->c3->b1->b2->b3->c1->c2->c3
链表B后接链表A:b1->b2->b3->c1->c2->c3->a1->a2->c1->c2->c3
b2和a1 到达相同位置,,满足 从等长位置,同时遍历。代码如下:
node* get_intersection_node(node* headA, node* headB) {
node* acur = headA, * bcur = headB;
while (acur != bcur) {
acur = acur == nullptr ? headB : acur->next;
bcur = bcur == nullptr ? headA : bcur->next;
}
return acur;
}
如果没有交点,遍历A和B长度之和后,循环会退出。
第一种解法,只有临时变量,所以空间复杂度为O(1) , 时间复杂度为n+n+n ,即O(n)
第二种解法,只有临时变量,所以空间复杂度为O(1) , 时间复杂度为n+n ,即O(n)