判断两个相交链表的交点

如图所示:
判断两个相交链表的交点_第1张图片
结点定义如下:

typedef struct node* ListNode;
struct node{
	int val;
	ListNode next;
}

Solution 1:两个栈

可以看到,虽然不相交的部分长度不同,不太好处理,但是相交的部分长度是相同的。只要我们倒着找到第一个两个链表结点不同的地方,它的前一个就是相交的结点。 所以我们先把两个链表分别入栈,然后同时弹出栈顶元素,找到第一个不相同的即可。

ListNode judge(ListNode headA, ListNode headB){
	stack<ListNode> stack1, stack2;
	
	ListNode p1 = headA, p2 = headB, pre = NULL; //pre记录当前节点的前一个
	while(p1 != NULL){
		stack1.push(p1);
		p1 = p1->next;
	}
	while(p2 != NULL{
		stack2.push(p2);
		p2 = p2->next;
	}
	
	while(stack1.top() == stack2.top()){
		pre = stack1.top();
		stack1.pop();
		stack2.pop();
	}
	
	return pre;
}

Solution 2:哈希表

相交的元素一定可以在哈希表中找到。

ListNode judge(ListNode headA, ListNode headB){
	map<ListNode, int> m;
	ListNode p1 = headA, p2 = headB;
	while(p1 != NULL){
		m[p1] = 1;
		p1 = p1->next;
	}
	while(p2 != NULL{
		if(m.find(p2) != m.end())  break;
		p2 = p2->next;
	}
	
	return p2;
}

Solution 3:双指针

我们将上图转换为如下图所示的样子:
判断两个相交链表的交点_第2张图片
我们设定两个指针:p1、p2。两个指针同步遍历各自的链表。当遍历到当前指针尾部时,将指针指向另一个链表的初始节点。 如此进行一轮之后,两个指针一定会同时到达交点处。如下图所示:判断两个相交链表的交点_第3张图片
可以看到,再一轮之后,两个指针一定会同时到达较短链表的起始位置。所以它们继续同步移动时,一定会同时走到交点处。

代码如下:

ListNode judge(ListNode headA, ListNode headB){
	ListNode p1 = headA, p2 = headB;
	while(p1 != p2){
		p1 = p1 == NULL ? headB : p1->next;
		p2 = p2 == NULL ? headA : p2->next;
	}
	return p1;
}

你可能感兴趣的:(leetcode)