一、题目
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。如果该链表长度小于k,请返回一个长度为 0 的链表。
二、题解
2.1 快慢指针
第一个指针先移动k步,然后第二个指针再从头开始,这个时候这两个指针同时移动,当第一个指针到链表的末尾的时候,返回第二个指针即可。
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pHead ListNode类
* @param k int整型
* @return ListNode类
*/
public ListNode FindKthToTail (ListNode pHead, int k) {
// write code here
// 快慢指针,fast比slow先走k个节点,再同步走,当 fast == null 时
// slow刚好指向倒数第k个节点
ListNode slow, fast;
slow = fast = pHead;
while (k-- != 0) {
if (fast == null) {
// 如果链表元素少于 k 个,则直接返回
return null;
}
fast = fast.next;
}
// 开始同步移动
while (fast != null) {
slow = slow.next;
fast = fast.next;
}
return slow;
}
}
2.2 用ArrayList
public ListNode FindKthToTail (ListNode pHead, int k) {
// write code here
List<ListNode> list = new ArrayList<>();
while(pHead != null){
list.add((pHead));
pHead = pHead.next;
}
int index = list.size() - k;
if(index < list.size() && index >= 0){
return list.get(index);
}
return null;
}
2.3 用栈
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;
}
// 判断栈的元素是否小于k
if (stack.size() < k){
return null;
}
//在出栈串成新的链表
ListNode firstNode = stack.pop();
while (--k > 0) {
// 将出栈的元素重新连接成为链表
ListNode temp = stack.pop();
temp.next = firstNode;
firstNode = temp;
}
return firstNode;
}
}
标签:ListNode,fast,链表,pHead,return,必刷,null,TOP101
From: https://blog.51cto.com/u_16244372/7967139