相交链表160
程序员文章站
2024-03-23 09:40:58
...
相交链表
题目描述:编写一个程序,找到两个单链表相交的起始节点。没有就返回null。
注意:
- 如果两个链表没有交点,返回 null.
- 在返回结果后,两个链表仍须保持原有的结构。
- 可假定整个链表结构中没有循环。
- 程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。
题解思路:从A链表第一个元素开始,去遍历B链表,找到一个相同元素后,同步遍历A和B链表,所有元素相同,即两个链表为相交链表,并返回同步遍历的起始节点。
struct ListNode *getIntersectionNode(struct ListNode *headA, struct ListNode *headB) {
struct ListNode *A,*B,*temp,*res;
A=headA;
int flag=1;
while(A)
{
B=headB;
while(B)
{
if(B->val==A->val)
{
break;
}
B=B->next;
}
temp=B;
while(B&&A)
{
if(A!=B)
{
flag=0;
break;
}
A=A->next;
B=B->next;
}
if(!B&&!A)
{
flag=1;
res=temp;
break;
}
A=A->next;
}
if(!flag)
{
res=NULL;
}
return res;
}
思考:感觉这题目有点问题,比如A:[1,2,3,4,0,5] B:[3,2,0,5] ,看到其他解法,先去掉长的链表的前一部分,让两个链表长度相同,然后同步遍历,返回第一个相同的节点,但是测试用例却没有涉及到我这种情况,然后看官方题解,发现默认没有我这种情况,所以个人感觉题目不是很严谨。
理解官方题解的要求,发现就是上面的要求。
int len(struct ListNode *H)
{
int count=0;
struct ListNode *p=H;
while(p)
{
count++;
p=p->next;
}
return count;
}
struct ListNode *getIntersectionNode(struct ListNode *headA, struct ListNode *headB) {
struct ListNode *a,*b,*resa,*resb,*res;
int len_A,len_B,max,min;
a=headA;
b=headB;
len_A=len(headA);
len_B=len(headB);
max = len_A > len_B ? len_A : len_B;
min = (max == len_A) ? len_B : len_A;
resa= (max==len_A) ? headA : headB;
resb= (max==len_A) ? headB : headA;
while(max>min)
{
resa=resa->next;
max--;
}
while(resa&&resb)
{
if(resa==resb)
{
break;
}
resa=resa->next;
resb=resb->next;
}
return resa;
}
上一篇: 7、Set集合与Map集合