返回單鏈表中第k個節點(leet簡單篇八百七十六題)
阿新 • • 發佈:2018-12-16
給定一個帶有頭結點 head 的非空單鏈表,返回連結串列的中間結點。
如果有兩個中間結點,則返回第二個中間結點。
示例 1:
輸入:[1,2,3,4,5]
輸出:此列表中的結點 3 (序列化形式:[3,4,5])
返回的結點值為 3 。 (測評系統對該結點序列化表述是 [3,4,5])。
注意,我們返回了一個 ListNode 型別的物件 ans,這樣:
ans.val = 3, ans.next.val = 4, ans.next.next.val = 5, 以及 ans.next.next.next = NULL.
示例 2:
輸入:[1,2,3,4,5,6]
輸出:此列表中的結點 4 (序列化形式:[4,5,6])
由於該列表有兩個中間結點,值分別為 3 和 4,我們返回第二個結點
拿到題我們可以想到最簡單的辦法是遍歷一遍單鏈表,得到長度後下一次遍歷只走到單鏈表的中間位置即可。然而我們這樣的做法效率非常的低。
所以我們可以使用倆個指標一個快指標一個慢指標,快指標你可以理解為一次走倆步,慢指標一次只走一步,這樣等快指標到了尾部,慢指標就剛好走到了中間位置,這樣就只遍歷了一遍。
程式碼比較簡單我們不寫了,現在假如問題進階為返回第k個,同樣我們可以使用遍歷一邊得到長度的辦法,但是效率不高,所以我們又可以使用快慢指標的辦法,這次讓快指標走k步,然後讓倆個指標同時前進,最後慢指標的位置就是要返回的位置
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k)
{
struct ListNode* slow = pListHead;
struct ListNode* fast = pListHead;
while(k--)
{
if(fast)
{
fast = fast -> next;
}
else
{
return NULL;
}
}
while(fast)
{
fast = fast -> next;
slow = slow -> next;
}
return slow;
}