LeetCode 817：链表组件(计数)

最新推荐文章于 2022-08-28 00:57:22 发布

程程程乘什么

最新推荐文章于 2022-08-28 00:57:22 发布

阅读量195

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/chengda321/article/details/102505074

版权

LeetCode 专栏收录该内容

59 篇文章 0 订阅

订阅专栏

这篇博客探讨了解决LeetCode 817问题的两种策略。第一种方法是建立图并进行深度优先搜索（DFS），虽然适用于所有图，但时间和空间复杂度较高，均为O(n)。第二种方法巧妙地利用了链表的特性，通过遍历链表，当遇到节点在给定集合中，但其后继节点不存在或不在集合内时，表明找到了一个组件的结尾，组件计数增加。

摘要由CSDN通过智能技术生成

解法一：常规解法，建图+DFS，时间复杂度O(n)+O(n)，空间复杂度因为需要存储图，所以是O(n)

这种方法是通解，对于所有图都适用。

/**
 * 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_set<int> f(G.begin(), G.end());
        unordered_map<int, vector<int>> g;
        //根据G中节点，从list中建立图g
        int u = head->val;
        while(head->next) {
            head = head->next;
            int v = head->val;
            if(f.count(v) && f.count(u)) {
                g[u].push_back(v);
                g[v].push_back(u);
            }
            u = v;
        }
        //DFS
        int ans = 0;
        unordered_set<int> visited;
        for(int u : G) {
            if(visited.count(u)) continue;
            ++ans;
            dfs(u, g, visited);
        }
        return ans;

    }
private:
    void dfs(int cur, unordered_map<int, vector<int>>& g, unordered_set<int>& visited) {
        if(visited.count(cur)) return; //当前节点在visited集合中计数不为0(为1),说明已经遍历过该节点
        visited.insert(cur);
        for(int next : g[cur]) { //int next 与 const int next均可以通过,为什么要加const?
            dfs(next, g, visited);
        }
    }
};

解法二：利用list的特性，将G中的点映射到list上的话，每一个组件内部都是线性连接的，外部应该是断开的，所以可以利用这种特性，遍历链表，当一个节点在G中，但它的下一个节点为null或者下一节点的值不在G中，则说明这是一个组建的末尾元素，ans++，再继续遍历

class Solution {
public:
    int numComponents(ListNode* head, vector<int>& G) {
        unordered_set<int> g(G.begin(), G.end());
        int ans = 0;
        while(head) {
            if(g.count(head->val) && (!head->next || !g.count(head->next->val)))
                ++ans;
            head = head->next;
        }
        return ans;
    }
};