今天做的三道题比较简单~
给你两个单链表的头节点 headA 和 headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表不存在相交节点,返回 null 。
题目数据 保证 整个链式结构中不存在环。
注意,函数返回结果后,链表必须 保持其原始结构 。
这道题注意思路,尽量简洁代码,对照官方答案,是我不配了……
先附上不优雅的代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode *getIntersectionNode(struct ListNode *headA, struct ListNode *headB) {
struct ListNode *s = (struct ListNode *)malloc(sizeof(struct ListNode)*1);
struct ListNode * a, * b,* a_x,* b_x;
a = headA;
b = headB;
int len = 0;
int flag = 1;
while(a != NULL && b != NULL)
{
a = a->next;
b = b->next;
}
while(a!= NULL)
{
len++;
flag = 1;
a = a->next;
}
while(b!= NULL)
{
len++;
flag = 0;
b = b->next;
}
a_x = headA;
b_x = headB;
while(len > 0)
{
if(flag == 0)
{
b_x = b_x->next;
}
else
{
a_x = a_x->next;
}
len--;
}
while(a_x != NULL)
{
if(a_x == b_x)
{
return a_x;
}
else
{
a_x = a_x -> next;
b_x = b_x -> next;
}
}
return NULL;
}
优雅代码在这里
思路:就是两个指针一个指a一个指b,a走完了重新指到b,b走完了指到a。如果不相交,那两个指针就是把两个链表都走了一遍,最后都指向空。如果相交,那交的那一部分长度相同,第二次在相交的时候一定能碰到。因为最终两指针走的一样远。
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
if (headA == nullptr || headB == nullptr) {
return nullptr;
}
ListNode *pA = headA, *pB = headB;
while (pA != pB) {
pA = pA == nullptr ? headB : pA->next;
pB = pB == nullptr ? headA : pB->next;
}
return pA;
}
};
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/intersection-of-two-linked-lists/solution/xiang-jiao-lian-biao-by-leetcode-solutio-a8jn/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
虽然官方代码简洁 我还是觉得我的好
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)