链表中倒数第k个结点

题目描述

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

 

使用两个指针，一个指向第一个节点，一个指向第k个节点，然后两个指针同时前进，直至后面的指针到达链表尾部

一定要注意特殊输入测试，下面代码没有第一个if判断，测试不通过，添加这条if判断后，测试通过。

要注意k是无符号整数，k-1 得 2^w - 1 ，是一个很大的整数

 

/*
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;
        
        ListNode* bef = pListHead;
        ListNode* aft = pListHead;
        
        while(--k){
            if(aft->next != nullptr)
                aft = aft->next;
            else
                //链表的节点数小于k
                return nullptr;
        }
        
        while(aft->next != nullptr){
            aft = aft->next;
            bef = bef->next;
        }
        
        return bef;
    }
};