题目
给定链表头结点 head
,该链表上的每个结点都有一个 唯一的整型值 。同时给定列表 nums
,该列表是上述链表中整型值的一个子集。
返回列表 nums
中组件的个数,这里对组件的定义为:链表中一段最长连续结点的值(该值必须在列表 nums
中)构成的集合。
示例 1:
输入: head = [0,1,2,3], nums = [0,1,3]
输出: 2
解释: 链表中,0 和 1 是相连接的,且 nums 中不包含 2,所以 [0, 1] 是 nums 的一个组件,同理 [3] 也是一个组件,故返回 2。
示例 2:
输入: head = [0,1,2,3,4], nums = [0,3,1,4]
输出: 2
解释: 链表中,0 和 1 是相连接的,3 和 4 是相连接的,所以 [0, 1] 和 [3, 4] 是两个组件,故返回 2。
提示:
- 链表中节点数为
n
1 <= n <= 104
0 <= Node.val < n
Node.val
中所有值 不同1 <= nums.length <= n
0 <= nums[i] < n
nums
中所有值 不同
思路
- 因为链表和列表中的元素唯一,所以将列表转化为哈希表
- 扫描链表,当链表中某个节点在哈希表中,向下探索直到当前元素不在哈希表中;形成一个组件;此时结果
+1
代码
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def numComponents(self, head: Optional[ListNode], nums: List[int]) -> int:
conponents = set(nums)
cur = head
ret = 0
while cur:
if cur.val not in conponents:
cur = cur.next
else:
while cur and cur.val in conponents:
cur = cur.next
ret += 1
return ret
复杂度
- 时间复杂度: O ( n ) O(n) O(n)
- 空间复杂度: O ( m ) O(m) O(m)