817. 連結串列元件

WrRan發表於2024-09-11
題目連結 817. 連結串列元件
思路 連結串列-思維題
題解連結 官方題解
關鍵點 題意可以轉換為:1. 節點的值在nums中且節點位於起始位置 2. 節點的值在nums中且前一個節點不在nums
時間複雜度 \(O(n)\)
空間複雜度 \(O(m)\)

程式碼實現:

class Solution:
    def numComponents(self, head: Optional[ListNode], nums: List[int]) -> int:
        nums = set(nums)
        in_set = False
        answer = 0
        while head:
            if head.val not in nums:
                in_set = False
            elif not in_set:
                in_set = True
                answer += 1
            head = head.next
        return answer

相關文章