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

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

题目描述

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

思路

定义两个指针

p2指针移动一格，k减1，直到 k == 0

情况1

p2没移动到最后一个元素，令 p2 = p2->next，此时再把p2和p1一起移动，直到 p2== NULL
倒数第k个元素为：p1 -> next，返回即可

情况2

此时 k == 链表的长度，当 k = 0 时，p2移动到最后一个元素，此时倒数第 k 个元素就是第一个元素，返回 p1 即可

代码

    ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
        ListNode* p1 = pListHead;
        ListNode* p2 = pListHead;

        while(k--)
        {
            if(p2->next != NULL)
                p2 = p2->next;
            else
                return NULL;
        }
        if(!p2){
            return p1;
        }else{
            p2 = p2->next;
            while(p2 != NULL)
            {
                p2 = p2->next;
                p1 = p1->next;
            }
        }

        return p1->next;
    }