链表中倒数最后k个结点

• 链表基础操作

题目描述

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

示例

输入：{1,2,3,4,5},2
返回值：{4,5}
说明：
返回倒数第2个节点4，系统会打印后面所有的节点来比较。

C++实现

/**
 * struct ListNode {
 *  int val;
 *  struct ListNode *next;
 *  ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param pHead ListNode类
     * @param k int整型
     * @return ListNode类
     */
    ListNode* FindKthToTail(ListNode* pHead, int k) {
        // write code here
        // 1. lenght of link
        // 2. location
        int size = length(pHead);
        if (size < k) {
            return nullptr;
        }

        ListNode *p = pHead;
        for (int i = 0; i < size-k; i++) {
            p = p->next;
        }
        return p;
    }

private:
    int length(ListNode* pHead) {
        int size = 0;
        for (ListNode *p = pHead; p; p = p->next) {
            size++;
        }
        return size;
    }
};

golang实现

package main
//import "fmt"
import . "nc_tools"
/*
 * type ListNode struct{
 *   Val int
 *   Next *ListNode
 * }
 */

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param pHead ListNode类 
 * @param k int整型 
 * @return ListNode类
*/
func FindKthToTail( pHead *ListNode ,  k int ) *ListNode {
    // write code here
    size := length(pHead)
    if size < k {
        return nil
    }

    p := pHead
    for i := 0; i < size-k; i++ {
        p = p.Next
    }
    return p
}

func length(pHead *ListNode) int {
    size := 0
    for p := pHead; p != nil; p = p.Next {
        size++
    }
    return size
}