leetcode 924. 尽量减少恶意软件的传播

题意:

在一些分散的图中,某些图中有某些节点是病原体,与这些节点在一个图中的所有节点都被感染了,现在求去掉一个原始病原体,使得最终所有图中感染的节点最少

思路:

bfs,然后找只有一个病原体的联通快的最大数量,因为只有单个病原体的联通块去掉这个病原体之后会减少被感染的节点数量。如果有多个病原体在一个联通块,那么即使去掉一个病原体,这个联通块最终也会被感染。此外,如果有多个病原体节点都满足条件,返回最小的那个。是最小,不是索引最小。

代码:

class Solution {
public:
int flag[400]; //记录是否为病原体
int f[400];//记录是否已经访问过
int minMalwareSpread(vector<vector<int>>& graph, vector<int>& initial) {
    memset(flag,0,sizeof(flag));
    memset(f,0,sizeof(f));
    int sz = initial.size();
    for(int i=0;i<sz;i++){
        flag[initial[i]] = 1;
    }
    int ma = 0,num = 0,cu = 0,all=0;
    sz = graph.size();
    for(int i=0;i<sz;i++){
        queue<int>q;
        if(!f[i]&&flag[i])q.push(i);
        num = 0;all = 0;
        while(!q.empty()) //每一次遍历都找一个联通块中的病原体数量,总数量,以及病原体的最小节点编号
        {
            int aa = q.front();q.pop();
            if(flag[aa]==1)num++;all++;
            f[aa] = 1;
            for(int j=0;j<sz;j++){
                if(graph[aa][j]==1){
                    if(!f[j])q.push(j);
                }
            }
        }
        sort(initial.begin(),initial.end());
       if(num==1){
        if(all>ma){
            ma = all;
            cu = i;
        }
        if(all==ma){
            cu = min(cu,i);
        }
       }
    }
    if(ma==0)return initial[0];
    else return cu;
}
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值