剑指offer - 链表中倒数最后k个结点

题目描述

输入一个链表，输出一个链表，该输出链表包含原链表中从倒数第k个结点至尾节点的全部节点。
如果该链表长度小于k，请返回一个长度为 0 的链表。

示例

输入：{1,2,3,4,5},1 
返回值：{5}

算法思路

思路一

最简单暴力的思路：

1. 先遍历一遍链表，求出链表的长度 len
2. 再遍历一遍链表，到 len - k的位置

思路二

采用双指针的思路：

1. 前置指针 first 和 后置指针second
2. 先让 first 走 k 步
3. 然后 first 和 second 一起动，直到 first到达末尾

代码实现

实现一

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 *   public ListNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param pHead ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    public ListNode FindKthToTail (ListNode pHead, int k) {
        // write code here
        int len = 0;
        ListNode p = pHead;
        //求出链表长度
        while(p!=null){
            len++;
            p = p.next;
        }
        //如果链表长度小于k，返回
        if(len<k){
            //此时p为空链表
            return p;
        }
        
        //指针要停止的位置，因为count从1开始，所以这里也加了1
        int stop = len - k + 1;
        int count = 1;
        p = pHead;
        while(count<stop){
            p = p.next;
            count++;
        }
        return p;
            
    }
}

实现二

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 *   public ListNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param pHead ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    public ListNode FindKthToTail (ListNode pHead, int k) {
        // write code here
        ListNode first = pHead;
        //先让first走k步
        for(int i=0;i<k;i++){
            if(first!=null){
                first = first.next;
            }else{
                //此时first为空链表，可以作为返回值返回
                return first;
            }
            
        }
        
        //两个指针共同移动
        ListNode second = pHead;
        while(first!=null){
            first = first.next;
            second = second.next;
        }
        
        return second;
        
    }
}