求鏈表的倒數第k個數值(親測有效,環境VS2010)

題目:輸入一個鏈表,輸出該鏈表中倒數第k個結點的值。爲符合大家的習慣的,本題目從1開始計數,級鏈表的尾結點是倒數第一個結點。

代碼如下:


#include <stdio.h>
#include <stdlib.h>
struct ListNode
{
	int m_pValue;
	ListNode* m_pNext;
};
/*輸入鏈表爲NULL或k爲0時都安全返回,魯棒性良好*/
ListNode* FindkthToTail(ListNode* pListHead,unsigned int k)
{
	if(pListHead==NULL||k==0)
		return NULL;
	ListNode* pFirst=pListHead;
	ListNode* pSecond=NULL;
	for(unsigned int i=0;i<k-1;i++)
	{
		if(pFirst->m_pNext!=NULL)
			pFirst=pFirst->m_pNext;
		else
			return NULL;
	}
	pSecond=pListHead;
	while(pFirst->m_pNext!=NULL)
	{
		pFirst=pFirst->m_pNext;
		pSecond=pSecond->m_pNext;
	}
	return pSecond;
}
int main()
{
	//ListNode* pListHead=NULL;//用於測試輸入爲NULL
	ListNode* pListHead=(ListNode *)malloc(sizeof(ListNode));
	ListNode* pNew=NULL;
	ListNode* pOld=pListHead;
	int input=0;//輸入
	for(int i=0;i<10;i++)//創建鏈表,給鏈表輸入0-9這10個數
	{
		pNew=(ListNode *)malloc(sizeof(ListNode));
		pNew->m_pValue=i;
		pNew->m_pNext=NULL;
		pOld->m_pNext=pNew;
		pOld=pNew;
	}
	scanf("%d",&input);
	ListNode* p=FindkthToTail(pListHead,input);
	if(p==NULL)
		return -1;
	printf("%d",p->m_pValue);
	return 0;
}

一般我們會從頭到尾遍歷鏈表,但此種方法時間複雜度爲O(n*n)。上面這個程序運用兩個指針來遍歷,時間複雜度爲O(n).
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章