链表倒数第k个结点

实现一个函数，查找链表的倒数第k个结点

思路:使用两个指针，第一个指针先指向第k个结点，然后两个指针同时移动，第一个指针移动到末尾的时候，第二个指针刚好位于倒数第k个位置

C++

.hpp文件

//
//  LinkedList.hpp
//  CPlusPlusProject
//
//
//
//微信公众号:程序员开发环境


#ifndef LinkedList_hpp
#define LinkedList_hpp

#include <stdio.h>

#endif /* LinkedList_hpp */

struct LinkNode {
    int val;
    LinkNode *next;
};

LinkNode *lastKNode(LinkNode *head, int k);

.cpp文件

//
//
//微信公众号:程序员开发环境


#include "LinkedList.hpp"
#include <iostream>

LinkNode *lastKNode(LinkNode *head, int k){
    if (head == NULL || k <= 0) {
        return NULL;
    }
    
    LinkNode *firstPoint = head;
    LinkNode *secondPoint = head;
    int index = 0;
    while (firstPoint->next != NULL && index < k - 1) {
        firstPoint = firstPoint->next;
        index++;
    }
    
    //链表长度小于k，直接return NULL
    if (index < k - 1) {
        return NULL;
    }
    
    while (firstPoint->next != NULL) {
        firstPoint = firstPoint->next;
        secondPoint = secondPoint->next;
    }
    return secondPoint;
}

main函数

#include <iostream>
#include "LinkedList.hpp"

int main(int argc, const char * argv[]) {
    int arr[] = {1, 2, 3, 4, 5};
    LinkNode *head = createLink(arr, 5);
    LinkNode *result = lastKNode(head, 3);
    if (result != NULL) {
        std::cout<<result->val<<std::endl;
    }
    
    return 0;
}