來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/middle-of-the-linked-list/
題目
給定一個頭結(jié)點為 head 的非空單鏈表,返回鏈表的中間結(jié)點。
如果有兩個中間結(jié)點,則返回第二個中間結(jié)點。
示例 1:
輸入:[1,2,3,4,5]
輸出:此列表中的結(jié)點 3 (序列化形式:[3,4,5])
返回的結(jié)點值為 3 。 (測評系統(tǒng)對該結(jié)點序列化表述是 [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]
輸出:此列表中的結(jié)點 4 (序列化形式:[4,5,6])
由于該列表有兩個中間結(jié)點,值分別為 3 和 4,我們返回第二個結(jié)點。
提示:
- 給定鏈表的結(jié)點數(shù)介于 1 和 100 之間。
思路
我們使用快慢指針的思路。
快指針走兩步,慢指針走一步,當快指針走到隊尾,則慢指針為中間節(jié)點。此處會有兩種情況:
1.ListNode的節(jié)點個數(shù)為奇數(shù);
2.ListNode的節(jié)點個數(shù)為偶數(shù)。
前提:定義指針為slow和fast。
第一種情況:
假設ListNode為:1-->2-->3-->4-->5
初始狀態(tài):slow和fast都指向1。
走第一步:
slow --> 2;
fast --> 3;
走第二步:
slow --> 3;
fast --> 5;
結(jié)束。
則中間節(jié)點為slow所指向的3.
第二種情況:
假設ListNode為:1-->2-->3-->4
走第一步:
slow --> 2;
fast --> 3;
走第二步:
slow --> 3;
fast.next.next為空,結(jié)束循環(huán)。
此時,slow指2,fast指向3.
根據(jù)題意,我們要返回節(jié)點3,則需要再多一個判斷:
if (fast.next != null) {
slow = slow.next;
}
代碼
public ListNode middleNode(ListNode head) {
if (head == null || head.next == null) {
return head;
}
ListNode slow = head;
ListNode fast = head;
while (fast.next != null && fast.next.next != null) {
fast = fast.next.next;
slow = slow.next;
}
if (fast.next != null) {
slow = slow.next;
}
return slow;
}
結(jié)果
