【set】个人练习-Leetcode-817. Linked List Components

Rstln2023-08-12 19:30

题目链接：https://leetcode.cn/problems/linked-list-components/description/

题目大意：给出一个vector<int> nums，其中有一些数字。再给出一个链表的头指针head，链表内的元素各不相同。如果链表中有某一段（长度大于等于1）的元素都在nums中出现过，那么就算一个component，求链表中的component的个数。

思路：【判断是否在nums中出现过】直接用set就好了，如果是STL的话，用count方法很方便。不过测试了一下后发现时间花得有点多，于是换成了数组。

使用两个布尔值：last表示【上一个元素】【是否在nums中出现过】，用flag表示【当前元素】【是否在nums中出现过】

当last == false && flag == true时，说明出现了一个新的component，结果加一
当last == true && flag == false时，说明出现了当前的component结束了
在判断之后都需要更新last

完整代码

cpp 复制代码

class Solution {
public:
    int numComponents(ListNode* head, vector<int>& nums) {
        bool nm[10001] = {0};
        for (auto x : nums) {
            nm[x] = true;
        }

        ListNode* ptr = head;
        bool last = false;
        int ret = 0;
        
        while (ptr) {
            bool flag = nm[ptr->val];

            if (flag != last) {
                if (last) {
                    last = false;
                }
                else {
                    ret++;
                    last = true;
                }
            }
            ptr = ptr->next;
        }

        return ret;
    }
};