码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode--Intersection of Two Linked Lists

时间:2015-01-16 11:18:52      阅读:110      评论:0      收藏:0      [点我收藏+]

标签:leetcode   linklist   双指针   

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.

Credits:
Special thanks to @stellari for adding this problem and creating all test cases.

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution 
{
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) 
    {
        if(headA==NULL || headB==NULL)
			return NULL;
		//Ensure the start point isn't in the cycle;
		//Create a cycle in head B;
		ListNode* temp = headB;
		while(temp->next!=NULL)
		{
			if(temp == headA)
				return headA;
			temp = temp->next;
		}
		ListNode* tailB = temp;
		temp->next = headB;
		ListNode* slow = headA;
		ListNode* fast = headA;
		while(fast!=NULL && fast->next!=NULL)
		{
			slow = slow->next;
			fast = fast->next;
			fast = fast->next;
			if(fast==slow)
				break;
		}
		if(fast == NULL || fast->next==NULL)
		{
			tailB->next = NULL;
			return NULL;
		}
		fast = headA;
		while(fast!=slow)
		{
			fast = fast->next;
			slow = slow->next;
		}
		tailB->next = NULL;
		return slow;
    }
};


LeetCode--Intersection of Two Linked Lists

标签:leetcode   linklist   双指针   

原文地址:http://blog.csdn.net/shaya118/article/details/42774681

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!