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

题目描述

输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。

例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

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

题解

遍历两遍，第一遍统计链表总长度，求的倒数节点的正向位置
第二遍，找到节点

1. python

class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        cur = head
        count = 0
        while cur:
            count += 1
            cur = cur.next
        cur = head
        count -= k 
        while count:
            cur = cur.next
            count -= 1
        return cur

2. C语言

struct ListNode* getKthFromEnd(struct ListNode* head, int k){
    struct ListNode *cur = head;
    int count = 0;
    while (cur != NULL) {
        count += 1;
        cur = cur -> next;
    }
    cur = head;
    count -= k;
    while (count != 0) {
        count -= 1;
        cur = cur -> next;
    }
    return cur;
}

3.C++

和C一样，就改了一个变量声明

class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        ListNode *cur = head;
        int count = 0;
        while (cur != NULL) {
            count += 1;
            cur = cur -> next;
        }
        cur = head;
        count -= k;
        while (count != 0) {
            count -= 1;
            cur = cur -> next;
        }
        return cur;
    }
};