1. 程式人生 > >返回單鏈表中第k個節點(leet簡單篇八百七十六題)

返回單鏈表中第k個節點(leet簡單篇八百七十六題)

給定一個帶有頭結點 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; }