求链表中倒数第k个结点

struct ListNode       //定义结点
{
	int m_Value;
	ListNode* m_pNext;
};

思路一个很自然的想法是从链表的尾端开始回溯k步,即可得到结果,但一般这种题目会限制链表为单向链表,所以这种思路行不通。

    一个可行的思路是:用两个指针都指向头结点,第一个指针先往前走k-1步,然后两指针同步向前走,直到第一个到达尾结点。

    考虑代码的鲁棒性的情况,代码如下:

ListNode* FindKthTail(ListNode* pHead,int k)
{
	if(pHead==NULL)           //指针为空
		return pHead;
	ListNode* pNodeA=pHead;
	ListNode* pNodeB=NULL;
	for(int i=0;i<k-1;i++)
	{
		if((pNodeA=pNodeA->m_pNext)==NULL)     //考虑链表结点小于k的情况并将pNodeA前移k-1位
			return NULL;	
	}
	pNodeB=pHead;
	while(pNodeA->m_pNext!=NULL)        //同步向前移动两指针,当pNodeA到尾时pNodeB即为倒数第k个结点
	{
		pNodeB=pNodeB->m_pNext;
		pNodeA=pNodeA->m_pNext;
	}
	return pNodeB;
}

猜你喜欢

转载自blog.csdn.net/qq_38506897/article/details/79447650