问题
假设两个单向链表在某个结点相交,成为一个单向链表,两个链表的表头节点是已知的,但是相交的结点未知,也就是说,它们相交之前各自的节点数未知,并且两个链表的节点数也可能不同,可以使用栈来找到两个链表的节点吗?
思路
1.创建两个栈,来分别存储两个链表节点
2.遍历第一个链表,节点压入栈
3.遍历第二个链表,节点入栈
4.这时,两个栈就包含了对应的链表的节点地址
5.比较两个栈的栈顶元素,节点地址
6.如果两者相等,那么弹出两个栈的栈顶元素并且保存,因为相等,保存其中一个即可
7.重复上述过程,知道一个栈为空,或者栈顶元素不相等
8.此时临时变量就是相交节点
9.返回临时变量的值
时间复杂度O(m+n) 空间复杂度O(m+n)
代码
public static ListNode getIntersectNode(ListNode head1,ListNode head2){
Stack<ListNode> stackOne = new StringStack();
Stack<ListNode> stackTwo = new StringStack();
if(head1 == null || head2 == null){
return new ListNode(0);
}
while (head1.getNext() != null){
stackOne.push(head1);
head1 = head1.getNext();
}
while (head2.getNext() != null){
stackOne.push(head2);
head2 = head2.getNext();
}
ListNode temp = new ListNode(0);
while (true){
if(stackOne.isEmpty()||stackTwo.isEmpty()){
break;
}
ListNode one = stackOne.pop();
ListNode two = stackTwo.pop();
if(one != two){
break;
}
temp = one;
}
return temp;
}