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

(这个问题与 尽量减少恶意软件的传播 是一样的，不同之处用粗体表示。)

在节点网络中，只有当 graph[i][j] = 1 时，每个节点 i 能够直接连接到另一个节点 j。

一些节点 initial 最初被恶意软件感染。只要两个节点直接连接，且其中至少一个节点受到恶意软件的感染，那么两个节点都将被恶意软件感染。这种恶意软件的传播将继续，直到没有更多的节点可以被这种方式感染。

假设 M(initial) 是在恶意软件停止传播之后，整个网络中感染恶意软件的最终节点数。

我们可以从初始列表中删除一个节点，并完全移除该节点以及从该节点到任何其他节点的任何连接。如果移除这一节点将最小化 M(initial)， 则返回该节点。如果有多个节点满足条件，就返回索引最小的节点。

我的代码 200ms

class Solution {
public:
   void bfs(int i,vector<vector<int>>& graph,unordered_set<int>&s,vector<int> &v,vector<int>& v_source)
    {
        queue<int>que;
        vector<int> visit(graph.size(),0);
        que.push(i);
        while(!que.empty())
        {
            int m=que.front();
            que.pop();
            if(visit[m]==1)continue;
            visit[m]=1;
            for(int k=0;k<graph[m].size();k++)
            {
                if(graph[m][k]==0||visit[k]==1||s.count(k)==1) continue;
                v_source[k]=i;
                que.push(k);
                v[k]++;
            }
        }
    }
    int minMalwareSpread(vector<vector<int>>& graph, vector<int>& initial) {
        int index=-1,min_m=0;
        vector<int> v(graph.size(),0);
        vector<int> v_source(graph.size(),-1);
        vector<int> v_time(graph.size(),0);
        unordered_set<int>s;
        sort(initial.begin(),initial.end());
        for(int i=0;i<initial.size();i++)
            s.insert(initial[i]);
        for(int i=0;i<initial.size();i++)
        {
            bfs(initial[i],graph,s,v,v_source);
        }
        for(int i=0;i<graph.size();i++)
            if(v[i]==1)
            {
                v_time[v_source[i]]++;
            }
        index=initial[0];
        for(int i=0;i<initial.size();i++)
        {
            int j=initial[i];
            if(v_time[j]>min_m)
            {
                min_m=v_time[j];
                index=j;
            }
        }
        return index;
    }
};

网上代码 200 ms

class Solution {
private:
	enum { MAX_NUM = 301 };
	//记t=Num[i]：t<2时：i被感染了t次
	//t>=2时：i为感染源节点，i独立感染了其他t-2个节点
	int Num[MAX_NUM];		
	int ID[MAX_NUM];		//谁感染过你(仅当Num[i]==1时 ID[i])
public:
	int minMalwareSpread(vector<vector<int>>& graph, vector<int>& initial) {
		int N = graph.size();	//节点数
		fill(Num, Num + N, 0);	//初始被感染次数为0
		for (auto v : initial) {
			Num[v] = 2;	//t>=2时：i为感染源节点
			//ID[v] = v;		//此语句逻辑成立但不需要
		}
		for (auto v : initial) {
			queue<int>Q; Q.push(v);	//从源头出发
			while (!Q.empty()) {
				int now = Q.front(); Q.pop();
				for (int i = 0; i < N; ++i) {
					if (graph[now][i]	//能通
						&& Num[i]<2		//感染次数少于2，且不是感染源（因感染源Num[i]==2）（最后只需要找仅感染1次的）
						&& ID[i] != v) {//本感染源v未感染过（避免重复）
						ID[i] = v;
						Num[i]++;
						Q.push(i);
					}
				}
			}
		}
		//统计仅被一个感染源感染的节点，其数量记在该感染源头上。
		for (int i = 0; i < N; i++){
			if (1 == Num[i]) {
				Num[ID[i]]++;
			}
		}
		int ans = -1, best = 0;
		for (int v : initial){
			if (Num[v] > best) {
				best = Num[v];
				ans = v;
			}
			else if (Num[v] == best && v < ans) {	//同样多取下标小者
				ans = v;
			}
		}
		return ans;
	}
};