1. 程式人生 > >Leetcode - 連結串列專題 - 876

Leetcode - 連結串列專題 - 876

給定一個帶有頭結點 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,我們返回第二個結點。

 

提示:

  • 給定連結串列的結點數介於 1 和 100 之間。

方法一:使用列表進行轉換

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution(object):
    def middleNode(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        data = []
        while head:
            data.append(head.val)
            head = head.next
        length = len(data) 
        temp = data[length/2:]
        ptr = ListNode(0)
        ptr1 = ptr
        for item in temp:
            ptr.next = ListNode(item)
            ptr = ptr.next
        return ptr1.next

方法二: 設定一個記數變數進行計數

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution(object):
    def middleNode(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        count = 0
        ptr = head
        while ptr:
            ptr = ptr.next
            count += 1
        if count == 1:
            return head
        count = count / 2
        for i in range(0, count-1):
            head = head.next
        return head.next