【2022-10-12每日一题】817. 链表组件[Medium]

2022-10-12每日一题：817. 链表组件

难度：Medium

标签：数组 、 哈希表 、 链表

给定链表头结点 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 中所有值 不同

方法一：哈希表+遍历

详细思路过程见官方题解，这里只做个人刷题记录，方便后续查询阅读

思路

此题需要计算组件的个数，只需在链表中计算有多少组件的起始位置即可。当一个节点满足以下条件之一时，它是组件的起始位置：

• 节点的值在数组 nums 中且节点位于链表起始位置；

• 节点的值在数组 nums 中且节点的前一个点不在数组 nums 中。

遍历链表，计算出满足条件的点的个数即可。因为需要多次判断值是否位于数组 nums 中，用一个哈希集合保存数组 nums 中的点可以降低时间复杂度。

代码

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func numComponents(head *ListNode, nums []int) (ans int) {
    set := make(map[int]struct{}, len(nums))
    for _, num := range nums {
        set[num] = struct{}{}
    }
    for inSet := false; head != nil; head = head.Next {
        if _, ok := set[head.Val]; !ok { // 不存在nums中，重置inSet标识
            inSet = false
        } else if !inSet {
            // 在nums中，答案加1， 在inSet改为false之前都在一个链表段中
            inSet = true 
            ans++
        }
    }
	return ans
}

复杂度分析

• 时间复杂度：O(n)，需要遍历一遍链表。

• 空间复杂度：O(m)，其中 m 是数组 nums 的长度，需要一个哈希集合来存储 nums 的元素。

LeetCode题库地址

• https://leetcode.cn/problems/linked-list-components