AXin说算法:找出链表中倒数第K个节点

面试官:告诉我怎么找出链表中倒数第K个节点!

AXin:从后往前走K步不就行了。

面试官:不好意思,我说的链表是单链表,只能从前往后走。就像下面这样!倒数第二个就是值为5的节点。

在这里插入图片描述

AXin:这。。。容我想想。

面试官:怎么,不会?

【片刻之后。。。】

AXin:没有的事儿,这难不倒我,用双指针呗,和你上次问我怎么判断链表是否有环差不多的思路。

面试官:不错啊小子,还会举一反三了。那你说说怎么解决。

AXin:咱们学小学生,看图说话。以找出倒数第3个节点为例:

在这里插入图片描述

AXin:要找出倒数第K个节点,只需让B指针先走K步,然后A和B一起走,直到B为NULL,A指向倒数第K个节点。

面试官:嗯~ 不错。代码怎么写?

【片刻之后。。。】

AXin:请您过目:

/**
 * 链表节点定义.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */

struct ListNode* getKthFromEnd(struct ListNode* head, int k){
    struct ListNode* A = head;
    struct ListNode* B = head;
    int i = 0;
    
    while (i < k) { // B先走K步
        B = B->next;
        i++;
    }

    while (B) { // A、B同时向后走
        B = B->next;
        A = A->next;
    }

    return A;
}

面试官:好!明天来上班!

在这里插入图片描述

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页