题目:输入两个链表,找出它们的第一个公共节点。链表节点定义如下:
struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
示例:
输入:intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
输出:Reference of the node with value = 8
输入解释:相交节点的值为 8 (注意,如果两个列表相交则不能为 0)。从各自的表头开始算起,链表 A 为 [4,1,8,4,5],链表 B 为 [5,0,1,8,4,5]。在 A 中,相交节点前有 2 个节点;在 B 中,相交节点前有 3 个节点。
思路:第一反应是暴力解法,但往往头脑中想到的第一种方法并不是最好的。根据题目示例,两个链表是单链表且公共节点之后的数值也相同不存在分叉问题,在本题中从后向前遍历不实际,所以想能否实现同时遍历两个指针一起到第一个公共节点,若一起到第一个公共节点,那就需要一起出发,那就需要比较链表长度,如不相等,则长的先出发,然后再一起出发。
实现思路:先想异常或特殊处理的问题:
①两个链表为空的情况
②两个链表等长
③不存在第一公共节点
实现:首先遍历得到两个链表长度;求出两个链表长度之差;再定义两个指针分别指向两个链表的头结点;长链表先走长度之差步数;同时遍历直到找到第一个公共节点返回。
C++代码实现:
class Solution {
public:
ListNode *getIntersectionNode(ListNode *pHead1, ListNode *pHead2) {
//得到两个链表长度
unsigned int nLength1 = GetListLength(pHead1);
unsigned int nLength2 = GetListLength(pHead2);
int nLengthDif = nLength1 - nLength2;
ListNode* pListHeadLong = pHead1;
ListNode* pListHeadShort = pHead2;
if(nLength2 > nLength1){
pListHeadLong = pHead2;
pListHeadShort = pHead1;
nLengthDif = nLength2 - nLength1;
}
//先在常链表上走nLengthDif步,再同时遍历;
for(int i=0;i<nLengthDif;i++){
pListHeadLong = pListHeadLong->next;
}
while((pListHeadLong != nullptr)&&(pListHeadShort != nullptr) && (pListHeadLong != pListHeadShort)){
pListHeadLong = pListHeadLong -> next;
pListHeadShort = pListHeadShort -> next;
}
ListNode* pFisrtCommonNode = pListHeadLong;
return pFisrtCommonNode;
}
unsigned int GetListLength(ListNode* pHead){
unsigned int nLength = 0;
ListNode* pNode = pHead;
while(pNode != nullptr){
++nLength;
pNode = pNode->next;
}
return nLength;
}
};