leetcode每日一题之相交链表
题目描述:
给你两个单链表的头节点 headA 和 headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表没有交点,返回 null 。
例子:
输入:intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
输出:Intersected at '8'
解释:相交节点的值为 8 (注意,如果两个链表相交则不能为 0)。
从各自的表头开始算起,链表 A 为 [4,1,8,4,5],链表 B 为 [5,0,1,8,4,5]。
在 A 中,相交节点前有 2 个节点;在 B 中,相交节点前有 3 个节点。
解法1:
使用Set集合(特点:元素不允许重复)存储节点,而不是存储节点的数值(起始自己的想法,发现遍历到1的时候就结束了),而存储节点的话,节点中是包含数值和下一个节点的指针指向的,因此才是正确的,具体代码如下:
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
Set<ListNode> nodeSet = new HashSet<>();//注意Set集合的泛型
while (headA != null) {//遍历A链表并添加到Set集合中
nodeSet.add(headA);
headA = headA.next;
}
while (headB != null) {//遍历B链表并做判断是否Set集合中的元素已经包含了B链表
if (nodeSet.contains(headB)) {
return headB;
} else {
headB = headB.next;
}
}
return null;
}
解法2:使用双指针
这种方法不容易想到,可以参考官方题解以及证明https://leetcode-cn.com/problems/intersection-of-two-linked-lists/solution/xiang-jiao-lian-biao-by-leetcode-solutio-a8jn/
大概意思就是当A链表遍历结束后将指针指向B链表的头节点,B链表也一样,假如相交部分的长度是c,A链表的长度是a,B链表的长度是b,那么两个链表走的路程都是a+b+c,之后会相遇在相交节点。
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
ListNode a = headA;
ListNode b = headB;
while (a != b) {
a = (a != null) ? a.next : headB;
b = (b != null) ? b.next : headA;
}
return a;
}