第一种方法:时间复杂度和空间复杂度都是O(N),直接用栈来实现
import java.util.*;
/*
* public class ListNode {
* int val;
* ListNode next = null;
* public ListNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pHead ListNode类
* @param k int整型
* @return ListNode类
*/
public ListNode FindKthToTail (ListNode pHead, int k) {
// write code here
if (pHead == null || k == 0) {
return null;
}
Stack<ListNode> stack = new Stack();
while (pHead != null) {
stack.push(pHead);
pHead = pHead.next;
}
ListNode node = null;
while (!stack.isEmpty()) {
node = stack.pop();
if (--k == 0) {
return node;
}
}
return null;
}
}
第二种方法:快慢指针的思想,先让快指针走k步,然后快慢指针同时走,让快指针走到终点n-k步(n为链表的长度),慢指针也走n-k步,也就是倒数k个节点,返回慢指针的位置即可,时间复杂度为O(n),空间复杂度为O(1)
import java.util.*;
/*
* public class ListNode {
* int val;
* ListNode next = null;
* public ListNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pHead ListNode类
* @param k int整型
* @return ListNode类
*/
public ListNode FindKthToTail (ListNode pHead, int k) {
// write code here
if (pHead == null || k == 0) {
return null;
}
ListNode fast = new ListNode(-1);
fast.next = pHead;
while (k > 0) {
fast = fast.next;
if (fast == null) {
return null;
}
k--;
}
ListNode slow = new ListNode(-1);
slow.next = pHead;
while (fast != null) {
fast = fast.next;
slow = slow.next;
}
return slow;
}
}