【C++】返回单链表的第k个结点

返回单链表的第k个结点，双指针可以很好地解决这个问题：

ListNode* FindKthToTail(ListNode* pListHead, unsigned k)
{
	ListNode *pAhead = pListHead;
	ListNode *pBehind = NULL;
	for(unsigned int i = 0; i < k - 1; i++)
	{
		pAhead = pAhead->next;
	}
	pBehind = pListHead;
	while( pAhead->next != NULL)
	{
		pAhead = pAhead->next;
		pBehind = pBehind->next;
	}
	return pBehind;
}

但是这段代码没有检查输入合法性，如果输入的是链表是NULL，程序崩溃；如果链表个数小于K，程序同样会因为在for循环里指向空指针而崩溃；如果输入的k为0，k-1得到将不是-1，而是无符号的0xFFFFFFFF,那么程序将循环这么多次，同样会崩溃。

考虑健壮性的代码应该是这样：

ListNode* FindKthToTail(ListNode* pListHead, unsigned k)
{
	if( pListHead == NULL || k == 0 ) return NULL;
	
	ListNode *pAhead = pListHead;
	ListNode *pBehind = NULL;
	
	for(unsigned int i = 0; i < k - 1; i++)
	{
		if(pAhead->next != NULL) 
		{
			pAhead = pAhead->next;
		}
	}
	pBehind = pListHead;
	while( pAhead->next != NULL)
	{
		pAhead = pAhead->next;
		pBehind = pBehind->next;
	}
	return pBehind;
}