点击查看剑指Offer全解【Java & Golang】实现
题目描述
输入一个链表,输出该链表中倒数第k个结点。
思路
第一种思路是将链表遍历一遍,将所有节点存储在list中,然后取倒数第K个节点即可。这种做法所需时间复杂度为O(n),空间复杂度为O(n)。
第二种思路是使用快慢指针,让fast指针快slow指针k个节点,那么在fast指针指向末尾null时,则说明slow指针当前已经指向了倒数第K个节点。这种做法所需的时间复杂度是O(n),空间复杂度为O(1)。
Java第一种思路实现
/*
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}*/
import java.util.ArrayList;
public class Solution {
public ListNode FindKthToTail(ListNode head,int k) {
ArrayList<ListNode> list = new ArrayList<>();
while (head != null) {
list.add(head);
head = head.next;
}
int size = list.size();
if (size == 0 || size - k < 0 || k <= 0) {
return null;
}
return list.get(size - k);
}
}
Java第二种思路实现
/*
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}*/
public class Solution {
public ListNode FindKthToTail(ListNode head,int k) {
ListNode slow = head;
ListNode fast = head;
// 先让fast快slow K步
for (int i = 0; i < k; i++) {
if (fast == null) {
return null;
}
fast = fast.next;
}
int step = 0;
// 记录fast走到null的步长,slow再走一遍步长即为倒数第k个节点
while (fast != null) {
fast = fast.next;
step++;
}
for (int i = 0; i < step; i++) {
slow = slow.next;
}
return slow;
}
}