剑指offer 22.链表中倒数第k个节点

题目描述

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

示例：

 给定一个链表: 1->2->3->4->5, 和 k = 2.

 返回链表 4->5.

第一种解法：

思路: 先循环得到链表的长度，再做判断

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
 
struct ListNode* getKthFromEnd(struct ListNode* head, int k){
    int i = 0,j;
    struct ListNode* p = head;
    if(head == NULL) {
        return NULL;
    }
    //遍历链表得到链表长度
    while(p != NULL) {
        p = p->next;
        i++;
    }
    p = head;
    //如果k大于链表长度，返回空
    if(i < k) {
        return NULL;
    }
    //k不大于链表长度，让 p 指向倒数第k个元素，返回p
    for(j = 0;j < i - k;j++) {
        p = p->next;
    }
    return p;
}

第一种解法简单粗暴，效率低，我们再介绍第二种解法

第二种解法：

思路： 第二种解法是采用双指针，一个快指针，一个慢指针。先让快指针走k步，指到第k个结点，再让慢指针和快指针一起走，当快指针指到尾部时，慢指针刚好指到倒数第k个结点。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */

struct ListNode* getKthFromEnd(struct ListNode* head, int k){
    struct ListNode* pfast = head;
    struct ListNode* pslow = head;
    if(head == NULL || k <= 0) {
        return NULL;
    }
    //用for循环写还要定义一个i，i也要占内存；所以换成while循环
    /*
    for(int i = 1; i < k; i++) {
    	if(pfast->next != NULL) {
            pfast =pfast->next;
        }else {
            return NULL;
        }
    }
    */
    k = k - 1;
    while(k--) {
    //判断一下k值会不会超出链表长度
        if(pfast->next != NULL) {
            pfast =pfast->next;
        }else {
            return NULL;
        }
    }
    while(pfast->next != NULL) {
        pfast = pfast->next;
        pslow = pslow->next;
    }
    return pslow;
}

两种解法的对比： 暴力循环解法花费4ms，双指针耗时0ms

题目及执行时长数据来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/lian-biao-zhong-dao-shu-di-kge-jie-dian-lcof