看到 LeetCode 上这个评论,30 个程序员想起了前任!
大家好,我是吴师兄。
我是万万没想到,一道平平无奇简简单单的算法题,暴露出程序员浪漫的秘密。
来源:https://leetcode-cn.com/problems/liang-ge-lian-biao-de-di-yi-ge-gong-gong-jie-dian-lcof/comments/
更有大情种,刷这道算法题都能想起前任!
或许很多人没有想到这一层,但看完评论勾起了一段回忆,情不自禁的点了个赞。
我们来看看是哪道题目有着这种魅力。
题目是剑指 Offer 52 号问题:两个链表的第一个公共节点,描述如下:
输入两个链表,找出它们的第一个公共节点。
题目清晰,来看分析。
无论 A、B 两个链表是否有相交点,最终都会指向一个相同的节点,要么是它们的公共尾部,要么是 NULL。
依托于上面这个结论,设置两个指针,pointA 和 pointB。
让指针 pointA 和 pointB 分别指向链表 A 和链表 B 的头结点,之后两个指针分别同时以步幅为 1 的速度向链表的尾部遍历。
当指针 pointA 遍历到链表 A 的尾节点时,此时指针 pointA 走了 a 个节点,将指针 pointA 指向链表 B 的头部,继续向后遍历,直至走到 c1,此时指针 pointA 总共走了 a + ( b - c ) 步。
当指针 pointB 遍历到链表 B 的尾节点时,此时指针 pointB 走了 b 个节点,将指针 pointB 指向链表 A 的头部,继续向后遍历,直至走到 c1,此时指针 pointB 总共走了 b + ( a - c ) 步。
根据数学知识,a + ( b - c ) = b + ( a - c ) 。
-
如果 c > 0,表明两链表有公共尾部, c1 存在,两两链表同时到达 c1;
-
如果 c = 0,表明两链表没有公共尾部,指针 pointA 和 pointB 都指向 NULL。
这也从数学逻辑上证明了:只要二人有交集,就终会相遇。
最后,给大家一份正经的代码:
// 登录 AlgoMooc 官网获取更多算法图解
// https://www.algomooc.com
// 作者:程序员吴师兄
// 代码有看不懂的地方一定要私聊咨询吴师兄呀
public class Solution {
public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
// 边界判断
if (headA == null || headB == null) return null;
// 设置一个指针 pointA,指向链表 A 的头节点
ListNode pointA = headA;
// 设置一个指针 pointB,指向链表 B 的头节点
ListNode pointB = headB;
// 指针 pointA 和 指针 pointB 不断向后遍历,直到找到相交点
// 不用担心会跳不出这个循环,实际上在链表 headA 长度和链表 headB 长度的之和减一
// pointA 和 pointB 都会同时指向 null
// 比如 headA 的长度是 7,headB 的长度是 11,这两个链表不相交
// 那么 pointA 移动了 7 + 11 - 1 次之后,会指向 null
// pointB 移动了 7 + 11 - 1 次之后,也指向 null
// 这个时候就跳出了循环
while (pointA != pointB) {
// 指针 pointA 一开始在链表 A 上遍历,当走到链表 A 的尾部即 null 时,跳转到链表 B 上
if( pointA == null){
// 指针 pointA 跳转到链表 B 上
pointA = headB;
}else{
// 否则的话 pointA 不断的向后移动
pointA = pointA.next;
}
// 指针 pointB 一开始在链表 B 上遍历,当走到链表 B 的尾部即 null 时,跳转到链表 A 上
if( pointB == null){
// 指针 pointA 跳转到链表 B 上
pointB = headA;
}else{
// 否则的话 pointB 不断的向后移动
pointB = pointB.next;
}
}
// 1、此时,pointA 和 pointB 指向那个相交的节点,返回任意一个均可
// 2、此时,headA 和 headB 不相交,那么 pointA 和 pointB 均为 null,也返回任意一个均可
return pointA;
}
}
是不是觉得很简单?
摸鱼时间又掌握了一道算法题!
最后,在这里依旧给自己打个广告,第五期算法训练营已经开营啦!