剑指offer 两个链表的第一个公共结点

题目:

输入两个链表,找出它们的第一个公共结点。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)

代码:

 1 struct ListNode {
 2     int val;
 3     struct ListNode *next;
 4     ListNode(int x) :
 5             val(x), next(NULL) {
 6     }
 7 };*/
 8 class Solution {
 9 public:
10     ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
11         ListNode* p1 = pHead1;
12         ListNode* p2 = pHead2;
13         while(p1 != p2){
14             if(p1 != NULL) p1 = p1->next;
15             if(p2 != NULL) p2 = p2->next;
16             if(p1 != p2){
17                 if(p1 == NULL) p1 = pHead2;
18                 if(p2 == NULL) p2 = pHead1;
19             }
20         }
21         return p1;
22     }
23 };

我的笔记:

假定 List1长度: a+n  List2 长度:b+n, 且 a<b

那么 p1 会先到链表尾部, 这时p2 走到 a+n位置,将p1换成List2头部

接着p2 再走b+n-(n+a) =b-a 步到链表尾部,这时p1也走到List2的b-a位置,还差a步就到可能的第一个公共节点。

将p2 换成 List1头部,p2走a步也到可能的第一个公共节点。如果恰好p1==p2,那么p1就是第一个公共节点。  或者p1和p2一起走n步到达列表尾部,二者没有公共节点,退出循环。 同理a>=b.

时间复杂度O(n+a+b)

posted @ 2020-06-09 21:06  John_yan15  阅读(107)  评论(0编辑  收藏  举报