[数据结构]链表组件（leetcode817）c++

链表组件（leetcode817）c++

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

    //G中所有的元素能构成多少个head中相连的子链表？
    //遍历链表，对于head中的每个节点 a 判断其 a->val 是否在G中存在，如果存在那么G中对应的 
    //a->val 很可能是一个组件；此时还得检查一下 a->next（假设为 b）, 如果 b->val 也在G中，
    //那么 (a->b)形成了一个更大的子链表，以b结尾的链表很可能是一个组件

    //算法：如果当前的节点在G 中，且下一个节点不在G 中，就找到了一个组件的尾节点，可以将答案加 1
    //首先创建一个SET，把G中元素复制到SET中
    //然后从头开始遍历链表，如果当前结点在SET中且下一个结点不在SET中就++
    //返回结果

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
 
#include <set>
class Solution {
public:
    int numComponents(ListNode* head, vector<int>& G) {
        //G中所有的元素能构成多少个head中相连的子链表？
        //遍历链表，对于head中的每个节点 a 判断其 a->val 是否在G中存在，如果存在那么G中对应的 
        //a->val 很可能是一个组件；此时还得检查一下 a->next（假设为 b）, 如果 b->val 也在G中，
        //那么 (a->b)形成了一个更大的子链表，以b结尾的链表很可能是一个组件

        //算法：如果当前的节点在G 中，且下一个节点不在G 中，就找到了一个组件的尾节点，可以将答案加 1
        //首先创建一个SET，把G中元素复制到SET中
        //然后从头开始遍历链表，如果当前结点在SET中且下一个结点不在SET中就++
        //返回结果

        //--[1]--
        std::set<int> node_set;
        for(int i=0;i< G.size();i++){
            node_set.insert(G[i]);
        }
        //~

        //--[2]--
        int res=0;
        ListNode *p;
        p=head;
        if(p==nullptr){
            return 0;
        }
        
        while(p){
            if(node_set.find(p->val)!=node_set.end()){
                if(p->next==nullptr || node_set.find(p->next->val)==node_set.end()){
                    res++;
                }
            }
            p=p->next;
        }
        //~

        //--[3]--
        return res;
        //~

        
    }
};