928. 尽量减少恶意软件的传播 II

最新推荐文章于 2024-07-22 09:38:37 发布

咔咔咔的

最新推荐文章于 2024-07-22 09:38:37 发布

阅读量178

点赞数 3

分类专栏： leetcode 文章标签： c++

本文链接：https://blog.csdn.net/weixin_45256307/article/details/138064945

版权

leetcode 专栏收录该内容

515 篇文章 2 订阅

订阅专栏

928. 尽量减少恶意软件的传播 II

题目链接：928. 尽量减少恶意软件的传播 II

代码如下：

class Solution {
public:
    int minMalwareSpread(vector<vector<int>>& graph, vector<int>& initial) {
        unordered_set<int> st(initial.begin(),initial.end());
        vector<bool> visit(graph.size(),false);
        int node_id,size;
        function<void(int)> dfs=[&](int x)
        {
            visit[x]=true;
            size++;
            for(int y=0;y<graph[x].size();y++)
            {
                if(graph[x][y]==0)  continue;
                if(st.contains(y))
                {
                    if(node_id!=-2&&node_id!=y)
                    {
                        node_id=node_id==-1?y:-2;
                    }
                }
                else if(!visit[y])
                {
                    dfs(y);
                }
            }
        };

        unordered_map<int,int> cnt;
        for(int i=0;i<graph.size();i++)
        {
            if(visit[i]||st.contains(i))    continue;
            node_id=-1;
            size=0;
            dfs(i);
            if(node_id>=0)
            {
                cnt[node_id]+=size;
            }
        }
        
        int max_cnt=0;
        int min_node_id=0;
        for(auto [node_id,c]:cnt)
        {
            if(c>max_cnt||c==max_cnt&&node_id<min_node_id)
            {
                max_cnt=c;
                min_node_id=node_id;
            }
        }
        retArn cnt.empty()?ranges::min(initial):min_node_id;
    }
};