题目:
给你两个单链表的头节点 headA 和 headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表没有交点,返回 null 。
问题:
- 判断两个链表是否相交?
- 如果相交,求交点。
思路:
思路1:暴力求解 - 穷举
依次取A链表中的每个节点跟B链表的所有节点比较,如果有地址相同的节点,就是相交,第一个相同的交点。
时间复杂度为O(N^2)
思路2:要求优化到O(N)
- 尾节点相同就是相交,否则就不想交。
- 求交点: 长的链表先走(长度差)步,再同时走,第一个相同就是交点。
代码:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode *getIntersectionNode(struct ListNode *headA, struct ListNode *headB) {
struct ListNode* TailA = headA;
struct ListNode* TailB = headB;
int lenA = 1;
while (TailA->next)
{
++lenA;
TailA = TailA->next;
}
int lenB = 1;
while (TailB->next)
{
++lenB;
TailB = TailB->next;
}
//要比较节点的地址,不要比较节点的值,否则容易误判
if (TailA != TailB)
{
return NULL;
}
//取绝对值
int gap = abs(lenA - lenB);
//先默认长链表和短链表
struct ListNode* longList = headA;
struct ListNode* shortList = headB;
if (lenA < lenB)
{
longList = headB;
shortList = headA;
}
//长的先走差距步,再同时走找交点
while (gap--)
{
longList = longList->next;
}
while (longList != shortList)
{
longList = longList->next;
shortList = shortList->next;
}
return longList;
}