# 817. 链表组件

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

# 题解

根据题意,我们只需要找出数组中的 在链表中是连续的数 的组数。

/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
int numComponents(ListNode* head, vector<int>& nums) {
int ans=0;
unordered_set<int> m;
for(int i=0;i<nums.size();i++){
m.emplace(nums[i]);
}
int flag=1;
while(head!=NULL){
if(flag&&m.count(head->val)){
flag=0;
head=head->next;
ans++;
}else if(!flag&&m.count(head->val)){
head=head->next;
}else{
flag=1;
head=head->next;
}
}
return ans;
}
};

复杂度分析

  • 时间复杂度:$O (n)$
  • 空间复杂度:$O (m)$