JZ22 链表中倒数最后k个结点

379 阅读1分钟

www.nowcoder.com/practice/88…

描述

输入一个长度为 n 的链表,设链表中的元素的值为 ai ,输出一个链表,该输出链表包含原链表中从倒数第 k 个结点至尾节点的全部节点。

如果该链表长度小于k,请返回一个长度为 0 的链表。

\

\

数据范围:0 \leq n \leq 10^50≤n≤105,0 \leq a_i \leq 10^90≤ai​≤109,0 \leq k \leq 10^90≤k≤109

要求:空间复杂度 O(n)O(n),时间复杂度 O(n)O(n)

进阶:空间复杂度 O(1)O(1),时间复杂度 O(n)O(n)

示例1

输入:

{1,2,3,4,5},3

复制

返回值:

{3,4,5}

示例2

输入:

{2},8

返回值:

{}

解法一:

使用快慢指针。第一个指针先移动k步,然后第二个指针再从头开始,这个时候这两个指针同时移动,当第一个指针到链表的末尾的时候,返回第二个指针即可。

时间复杂度O(N):N为链表长度,遍历整个链表

空间复杂度O(1):使用额外常数大小空间

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 quickNode = pHead;
        ListNode slowNode = pHead;

        for (int i = 0; i < k; i++) {
            if (quickNode == null) {
                return null;
            } else {
                quickNode = quickNode.next;
            }
        }

        while (quickNode != null) {
            slowNode = slowNode.next;
            quickNode = quickNode.next;
        }

        return slowNode;
    }
}

解法二:

使用栈。把原链表的结点全部压栈,然后再把栈中最上面的k个节点出栈,出栈的结点重新串成一个新的链表即可。

时间复杂度O(N):N表示链表的数量,遍历链表

空间复杂度O(N):存储链表元素的栈空间

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;
    }
}