1.暴力
我们可以分别算出两个链表的长度,然后将长链表的指针指向和短指针距离尾部相同距离的位置,两个指针同时开始往后移动,看是否移到相同的节点
代码如下:
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
int len_a,len_b;
len_a=len_b=0;
ListNode *a=headA,*b=headB;
while(a!=NULL)
{
len_a++;
a=a->next;
}
while(b!=NULL)
{
len_b++;
b=b->next;
}
a=headA;
b=headB;
while(len_a>len_b)
{
len_a--;
a=a->next;
}
while(len_a<len_b)
{
len_b--;
b=b->next;
}
while(a!=NULL && b!=NULL)
{
if(a==b)
{
return a
}
a=a->next;
b=b->next;
}
return NULL;
}
};
结果是这样:
2.或许减小一点时间复杂度常量系数的做法
对上面第一个例子,我们可以设置两个指针,分别指向两个头,一起向后移动,将第一个到头的指针指向另外一个链表头部,画一画,其实这样就统一了步调,代码如下所示:
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
ListNode *ptrA=headA,*ptrB=headB;
while(ptrA!=ptrB)
{
ptrA=(ptrA==NULL)?headB:ptrA->next;
ptrB=(ptrB==NULL)?headA:ptrB->next;
}
if(ptrA==ptrB) return ptrA;
return NULL;
}
};
是不是少了很多代码看着很舒服?嘿嘿~
之前有一次写成这样了:
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
ListNode *ptrA=headA,*ptrB=headB;
while(ptrA!=ptrB)
{
ptrA=(ptrA->next==NULL)?headB:ptrA->next;
ptrB=(ptrB->next==NULL)?headA:ptrB->next;
}
if(ptrA==ptrB) return ptrA;
return NULL;
}
};
不过按照个别例子[2,6,4],[1,5]可以看出,这个其实是没有求好两个链表之间的长度差异的。
有一样错误的同学可以一步一步试着做做看...