欢迎来到王正伟的博客

返回顶部

面试题22:链表中的倒数第k个节点(C++)

题目地址:https://leetcode-cn.com/problems/lian-biao-zhong-dao-shu-di-kge-jie-dian-lcof/

题目描述

输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。

题目示例

示例:

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

解题思路

思路1:最容易想到的是对链表遍历统计链表长度,然后减去k,得到链表倒数第k个位置,最后遍历输出即可。

思路2:快慢双指针,使用双指针方法不需要统计链表长度,定义指针high表示向前走k步,low指向倒数第k个节点,其中,low与high相距k步,初始化时,两个指针均指向头节点head,需要注意的是,当high指针走到链表的尾节点并跳出后,指针low与尾节点距离为k-1,即low指向倒数第k个节点。

图1

图2

 图3

程序源码

普通方法

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        if(head == nullptr) return nullptr;
        ListNode* p = head;
        int cnt = 0;
        while(p)
        {
            p = p->next;
            cnt++;
        }
        for(int i = cnt - k; i > 0; i--) //前进cnt-k步找到倒数第k个节点,即i=0;i<cnt-k;i++
        {
            head = head->next; //遍历找到节点4,
        }
        return head; //返回倒数第2个节点4,以及链表中中还未遍历的节点5
    }
};

快慢指针

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        if(head == nullptr) return nullptr;
        ListNode* low = head, *high = head;
        for(int i = 0; i < k; i++)
        {
            if(high == nullptr) return nullptr; //high走到链表尾节点并跳出
            high = high->next;
        }
        while(high != nullptr)
        {
            high = high->next;
            low = low->next;
        }
        return low;
       }
};

参考文章

https://leetcode-cn.com/problems/lian-biao-zhong-dao-shu-di-kge-jie-dian-lcof/solution/mian-shi-ti-22-lian-biao-zhong-dao-shu-di-kge-j-11/

posted @ 2020-05-07 15:03  Mr.King~  阅读(305)  评论(0编辑  收藏  举报