算法记录
LeeCode 题目:
输入两个链表,找出它们的第一个公共节点。
说明
一、题目
输入两个链表,找出它们的第一个公共节点。
- 如果两个链表没有交点,返回 null。
- 在返回结果后,两个链表仍须保持原有的结构。
- 可假定整个链表结构中没有循环。
- 程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。
二、分析
- 两条链表如果能够相交,即从相交点之后的节点都是一致的。
- 如果链表能够简化成长度相等的链表,就可以直接对两条链表进行同时遍历,如果当前遍历节点相等则意味着链表相交,直接返回当前的节点即可;遍历结束必然会返回一个
null
节点。 - 为了使链表的长度一致,可以先分别遍历两条链表得到对应的长度,让长链表先走,知道剩余的长度和短链表相等停下来,就可以开始相等长度链表的判断了。
- 要注意判断是否同一节点的条件是节点是否相等,而不是节点的
val
值相等。
public class Solution {
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
ListNode A = headA;
ListNode B = headB;
int a = 0;
int b = 0;
while(A != null) {
A = A.next;
a++;
}
while(B != null) {
B = B.next;
b++;
}
if(a > b) {
for(int i = 0; i < a - b; i++)
headA = headA.next;
} else {
for(int i = 0; i < b - a; i++)
headB = headB.next;
}
while(headA != null && headB != null) {
if(headA == headB)
return headB;
headB = headB.next;
headA = headA.next;
}
return null;
}
}
总结
熟悉链表的表现和操作,会遍历计算链表的长度。