问题来源:链表的中间节点
问题描述:给定一个带有头结点 head 的非空单链表,返回链表的中间结点。如果有两个中间结点,则返回第二个中间结点。
例子:
输入:[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.
分析:这个例子其实很简单,直观的思路是两次遍历,第一遍遍历取得链表长度,确定middle所处位置,第二遍遍历直接取得middle节点
1. 两次遍历
class Solution:
def middleNode(self, head: ListNode) -> ListNode:
# 第一次遍历,确定middle位置
curNode = ListNode(0)
curNode.next = head
index = 0
while curNode.next:
index += 1
curNode = curNode.next
middle = index // 2 + 1
# 第二次遍历
curNode = ListNode(0)
curNode.next = head
index = 0
while curNode.next:
index += 1
if index == middle:
return curNode.next
curNode = curNode.next
2. 一次遍历
用一个技巧,快慢指针。快指针每次前进两个节点,慢指针每次前进一个节点,这样快指针走到链表末尾,慢指针走到一半节点。
class Solution:
def middleNode(self, head: ListNode) -> ListNode:
slow = ListNode(0)
fast = ListNode(0)
slow.next = head
fast.next = head
while fast:
if fast.next:
fast = fast.next.next
else:
fast = fast.next
slow = slow.next
return slow