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

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

示例：

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

解题思路

第一次遍历找出总的长度，第二次遍历输出倒数第k个节点

代码展示

/**
 * 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) {
        ListNode* h = head;
        int sum = 0;
        while(h!=NULL)
        {
            sum++;
            h=h->next;
        }

        for(int i=0;i<sum-k;i++)
        {
            head=head->next;
        }

        return head;
    }
};

优化代码

仅用了一次循环，利用两个指针找出倒数第k个结点。

class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        ListNode* h = head;
        int sum = k;
        while(h!=NULL)
        {
            if(!sum)head=head->next;
            else sum--;
    
            h=h->next;
        }

        return head;
    }
};