問題
輸入一個單向鏈表,輸出該鏈表中倒數第k個節點,鏈表的倒數第0個節點爲鏈表的尾指針。
解答
設置兩個指針 fast、slow,首先 fast 和 slow 都指向 head,然後 fast 向前走 k 步,這樣 fast 和 slow 之間就間隔 k 個節點,最後 fast 和 slow 同時向前移動,直至 fast 走到鏈表末尾, 這是slow指向的就是第k個節點。
代碼如下:
Node* theKthNode(Node *head,int k)
{
if(k < 0) return NULL; //異常判斷
Node *slow,*fast;
slow = fast = head;
int i = k;
for(;i>0 && fast!=NULL;i--)
{
fast = fast->next;
}
if(i > 0) return NULL; //考慮k大於鏈表長度的case
while(fast != NULL)
{
slow = slow->next;
fast = fast->next;
}
return slow;
}
注意該題中使用了雙指針來遍歷鏈表,雙指針策略可以解決很多經典的關於鏈表的問題,如鏈表是否有環。