【链表遍历】817. Linked List Components

最新推荐文章于 2020-11-23 10:18:10 发布

原创最新推荐文章于 2020-11-23 10:18:10 发布 · 370 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

107 篇文章

订阅专栏

本文介绍了一种遍历链表并统计特定元素组所形成独立组件数量的方法。使用C++实现，通过查找链表中是否包含给定集合G中的元素，并确保连续出现的集合元素被视为一个整体。

遍历链表，查询当前元素是否在G内；

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    int numComponents(ListNode* head, vector<int>& G) {
        unordered_map<int,bool> table;
        for(int i = 0;i<G.size();i++)
            table[G[i]]=1;
        
        int ans=0,flag=0;
        while(head!=NULL)
        {
            if(table.find(head->val)!=table.end())
                flag=1;
            else
            {
                if(flag==1)
                {
                    ans++;
                    flag=0;
                }
            }
            head=head->next;
        }
        if(flag==1)
            ans++;
        return ans;
    }
};