0
点赞
收藏
分享

微信扫一扫

面试必刷TOP101:8、链表中倒数最后k个结点

一、题目

输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。如果该链表长度小于k,请返回一个长度为 0 的链表。

面试必刷TOP101:8、链表中倒数最后k个结点_出栈

二、题解

2.1 快慢指针

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

面试必刷TOP101:8、链表中倒数最后k个结点_链表_02

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

举报

相关推荐

0 条评论