剑指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)