剑指offer最优解Java版-链表中倒数第k个结点
2019-06-23 本文已影响5人
全菜工程师小辉
题目描述
输入一个链表,输出该链表中倒数第k个结点。
第一种方案
先遍历链表获取长度,然后再次遍历链表输出倒数第k个结点。
class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}
public class Solution {
public ListNode FindKthToTail(ListNode head, int k) {
if (head == null) {
return null;
}
int count = 1;
ListNode cur = head;
while (cur.next != null) {
count++;
cur = cur.next;
}
cur = head;
if (k > count) {
return null;
}
for (int i = 0; i < count - k; i++) {
cur = cur.next;
}
return cur;
}
}
复杂度分析:
- 时间复杂度:O(n)。
- 空间复杂度:O(1)。
第二种方案
两个指针,先让第一个指针和第二个指针都指向头结点,然后再让第一个指正走(k-1)步,到达第k个节点。然后两个指针同时往后移动,当第一个结点到达末尾的时候,第二个结点所在位置就是倒数第k个节点了。
class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}
public class Solution {
public ListNode FindKthToTail(ListNode head, int k) {
if (head == null || k <= 0) {
return null;
}
ListNode pre = head;
ListNode last = head;
for (int i = 1; i < k; i++) {
if (pre.next != null) {
pre = pre.next;
} else {
return null;
}
}
while (pre.next != null) {
pre = pre.next;
last = last.next;
}
return last;
}
}
复杂度分析:
- 时间复杂度:O(n)。
- 空间复杂度:O(1)。