方法一 弹栈 笔试做法
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
stack<ListNode*>st;
ListNode* result=NULL;
while(head){
st.push(head);
head=head->next;
}
while(k--){
result=st.top();
st.pop();
}
return result;
}
};
解题思路
把所有的元素压入栈中,再弹出k个即可得到目标节点,或者也可以加到vector中,找到第n-k+1个位置,也是这个数
##方法二 快慢指针 面试做法
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
ListNode* slow=head;
ListNode* fast=head;
while(k--){
fast=fast->next;
}
while(fast){
fast=fast->next;
slow=slow->next;
}
return slow;
}
};
解题思路
使用两个指针,让某个指针先走k个元素,然后再让两个指针同时走,先走的那个指针到达尾部的时候,后走的那个指针正好位于倒数k个位置,本方法时间复杂度是O(N),但空间复杂度是O(1)