leetcode 817. 链表组件

1. 题目链接 https://leetcode-cn.com/problems/linked-list-components/submissions/

2. 题目描述

   1. 给定一个链表（链表结点包含一个整型值）的头结点 head。

      同时给定列表 G，该列表是上述链表中整型值的一个子集。

      返回列表 G 中组件的个数，这里对组件的定义为：链表中一段最长连续结点的值（该值必须在列表 G 中）构成的集合。

   2. 示例 1：

      输入: 
      head: 0->1->2->3
      G = [0, 1, 3]
      输出: 2
      解释: 
      链表中,0 和 1 是相连接的，且 G 中不包含 2，所以 [0, 1] 是 G 的一个组件，同理 [3] 也是一个组件，故返回 2。

      示例 2：

      输入: 
      head: 0->1->2->3->4
      G = [0, 3, 1, 4]
      输出: 2
      解释: 
      链表中，0 和 1 是相连接的，3 和 4 是相连接的，所以 [0, 1] 和 [3, 4] 是两个组件，故返回 2。

3. 解题思路

   1. 初始时假设G内每个元素都是一个组件，即组件数为len(G)
   2. 遍历链表，当链表内元素存在于G中时，说明找到一个组件，继续遍历找到该组件的所有元素,找到一个元素组件数就减一

4. 代码

   1. python

      class Solution:
          def numComponents(self, head: ListNode, G) -> int:
              ans = len(G)
              G = set(G)
              while head:
                  if head.val in G:
                      head = head.next
                      while head and head.val in G:
                          ans , head = ans + 1, head.next
                  head = head.next if head else head
              return ans