題目:輸入一個鏈表,輸出該鏈表中倒數第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).