题目:
设置两个指针遍历两个链表节点,要找到两个链表相交的起始节点需要让两个链表尾部对齐,然后先让长的链表的指针遍历到与短链表对齐的位置,然后同时移动并判断两个指针是否相等。这种操作如下图所示:
具体的代码实现如下所示:
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
ListNode* curA = headA;
ListNode* curB = headB;//设置两个指针遍历两个链表
int lenA=0;
int lenB=0;//设置两个变量记录链表长度
while(curA != NULL) {
curA = curA->next;
lenA++;
}
while(curB != NULL) {
curB =curB->next;
lenB++;
}//计算两个链表的长度
curA = headA;
curB = headB;
if(lenB > lenA) {
swap(lenA, lenB);
swap(curA, curB);
}//保证A永远记录的是较长链表的信息
int gap = lenA - lenB;
while(gap-- && curA != NULL) {
curA = curA->next;
}//实现上文说的操作,即两链表尾部对齐,找到短链表开始的位置
while(curA != NULL) {
if(curA == curB) {
return curA;
}
curA = curA->next;
curB = curB->next;
}//一步一步移动寻找相交的节点,如果找到则返回,否则继续移动
return NULL;
}
};
运行结果如下所示: