LeetCode 3069: Distribute Elements Into Two Arrays I

基本資料

#

難易度: easy 第一次嘗試：2024-03-03

• 提交紀錄：Accepted（一次通過）

解題思路

#

直接照規則模擬（Array 模板五）：arr1 = [nums[0]]、arr2 = [nums[1]]，之後每個數看 arr1[-1] > arr2[-1] 決定放哪邊，最後接起來。

我的解法

#

class Solution:
    def resultArray(self, nums: List[int]) -> List[int]:
        a, b = [nums[0]], [nums[1]]
        for x in nums[2:]:
            (a if a[-1] > b[-1] else b).append(x)
        return a + b

複雜度

#

時間 O(n)，空間 O(n)。

要特別避免的錯誤

#

• 比的是兩個陣列的最後一個元素，不是總和。

面試口說整理

#

我照題目的規則模擬，每個數放到最後一個元素比較大的那個陣列，最後把兩個陣列接起來。時間 O(n)。