剑指offer 链表中倒数第k个结点

题目描述

输入一个链表，输出该链表中倒数第k个结点。

思路：设置两个指针，一个指针从链表第一个结点开始，另一个指针从第k个结点开始，两个指针同时走，当第二个指针到达链表尾时，第一个指针指向倒数第k个结点。

考虑特殊情况：链表长度小于k，那么返回空。

btw，通过测试，倒数第0个结点看做链表的最后一个结点。

/*
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) :
            val(x), next(NULL) {
    }
};*/
class Solution {
public:
    ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
        ListNode *tmp = pListHead, *res = pListHead;
        /* unsigned int cnt = 0;
        for (; tmp != NULL; cnt++) {
            if (cnt >= k)
                res = res->next;
            tmp = tmp->next;
        }
        return cnt < k ? NULL : res; */
        int cnt = 0;
        while (tmp != NULL && (cnt < k)) {
            tmp = tmp->next;
            cnt++;
        }
        while (tmp != NULL) {
            tmp = tmp->next;
            res = res->next;
        }
        return cnt < k ? NULL : res;
    }
};

 更新：根据剑指offer原书，最后一个节点，k=1，因此无符号整型k==0是要排除。

/*
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) :
            val(x), next(NULL) {
    }
};*/
class Solution {
public:
    ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
        if (pListHead == nullptr || k == 0) {
            return nullptr;
        }
        int cnt = 0;
        ListNode *low = pListHead, *fast = pListHead; 
        while (fast != nullptr && cnt < k) {
            fast = fast->next;
            cnt++;
        }
        while (fast != nullptr) {
            low = low->next;
            fast = fast->next;
        }
        return cnt == k ? low : nullptr;
    }
};

相关题目：求链表的中间节点。如果链表总长度为奇数，则返回中间节点，否则返回中间的任意一个节点。

思路：设置快慢指针，快指针每次走两步，慢指针每次走一步。

ListNode* FindMidNode(ListNode* pListHead) {
    if (pListHead == NULL) {
        return NULL;
    }
    ListNode *low = pListNode, *fast = pListNode;
    while (fast->next != NULL && fast->next->next != NULL) {
        low = low->next;
        fast = fast->next->next;
    }
    return low;
}