Leetcode-每日一题【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 中所有值 不同

解题思路

1.我们可以先设置一个boolean数组来存放nums列表中唯一的整型值，将nums中对应的整型值在Boolean数组中对应下标处将其赋值为1。

2.我们定义两个整型变量 cur 和 num，cur来记录是否存在组件 ，num用来记录组件个数

3.之后我们定义一个新结点来遍历链表，判断链表中的值是否在对应的boolean数组中为1，若为1，我们另cur++，若不是1我们判断cur是否为0，若是0则进行下一次循环，若不是我们另num++，也就是说我们已经找到了一个组件

4若链表最后一个元素也是组件，但是在判断完最后一个元素后会跳出循环，所以我们要再次对cur进行判断，将结果加入num，并return num；

代码实现

class Solution {
    public int numComponents(ListNode head, int[] nums) {
        boolean[] set = new boolean[10001];
        for(int i : nums){
            set[i] = true;
        }
         
        int cur = 0;
        int num = 0;
        for(ListNode i = head; i != null; i = i.next){
            if(set[i.val]){
                cur++;
            }else{
                if(cur > 0){
                    num++;
                    cur = 0;
                }
            }
        }
        return num + (cur > 0 ? 1 : 0);
    }

}

测试结果