Write a program to find the node at which the intersection of two singly linked lists begins.
For example, the following two linked lists:
A: a1 → a2 ↘ c1 → c2 → c3 ↗ B: b1 → b2 → b3
begin to intersect at node c1.
Notes:
- If the two linked lists have no intersection at all, return
null
. - The linked lists must retain their original structure after the function returns.
- You may assume there are no cycles anywhere in the entire linked structure.
- Your code should preferably run in O(n) time and use only O(1) memory.
找链表交界,很类似Linked List Cycle II那题,方法也是类似的双指针相遇法。分两步走:
1. 如何判断两链表是否相交?
两链表相交则他们必然有共同的尾节点。所以遍历两两链表,找到各自的尾节点,如果tailA!=tailB则一定不相交,反之则相交。
2. 如何判断两链表相交的起始节点?
在第1步判断相交时可以顺带计算两链表的长度lenA和lenB。让长的链表的head先走abs(lenA-lenB)步,然后和短链表的head一起走,直到两者相遇,即为要找的节点。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | class Solution { public: ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) { if(!headA || !headB) return NULL; int lenA = 0, lenB = 0; ListNode *tailA = headA, *tailB = headB; while(tailA->next) { tailA = tailA->next; lenA++; } while(tailB->next) { tailB = tailB->next; lenB++; } if(tailA!=tailB) return NULL; if(lenA>lenB) { for(int i=0; i<lenA-lenB; i++) headA = headA->next; } else { for(int i=0; i<lenB-lenA; i++) headB = headB->next; } while(headA!=headB) { headA = headA->next; headB = headB->next; } return headA; } }; |
No comments:
Post a Comment