相交链表
- 题目
- 思路
- C++代码
题目
给你两个单链表的头节点 headA 和 headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表不存在相交节点,返回 null 。
图示两个链表在节点 c1 开始相交:
思路
求两个链表的相交结点,使用双指针法,求出两个链表的长度的差值gap,让长的那一个链表先走gap步,使两个链表对齐,然后同时移动两个链表,直到找到相等的结点为止。
C++代码
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
ListNode* curA = headA;
ListNode* curB = headB;//用于遍历的指针
int lenA = 0;
int lenB = 0;//两个链表的长度
// 求链表A的长度
while(curA != NULL) {
lenA++;
curA = curA->next;
}
// 求链表B的长度
while (curB != NULL) {
lenB++;
curB = curB->next;
}
curA = headA;//返回开始位置
curB = headB;
if(lenA >= lenB){
int gap = lenA -lenB;//两个链表的长度的差值
while(gap--) curA = curA->next;
while(curA != NULL){
if (curA == curB) {
return curA;
}
curA = curA->next;
curB = curB->next;
}
}
if(lenA < lenB){
int gap = lenB - lenA;
while(gap--) curB = curB->next;
while(curB != NULL){
if (curA == curB) {
return curB;
}
curA = curA->next;
curB = curB->next;
}
}
return NULL;
}
};