2020-5 leetcode 928. 尽量减少恶意软件的传播 II

1.这道题和上一道题的差别在于，这道题删除节点是真的把该节点移除，同时删除原本的连接关系。而上一道题只是把感染的点变为不被感染的点。但原先的连接关系不变。

2.所以在这道题中，要找的点是 只能受这个点影响而被感染的点的个数最多的那个感染点。
3. 方法：对于不在intianal 中的点重新建一个图，然后DFS遍历。

class Solution {
public:
    int minMalwareSpread(vector<vector<int>>& graph, vector<int>& initial) {
        int n=graph.size();
        vector<int> clean(n,1);
        for(int x:initial) clean[x]=0;

        vector<vector<int>> infetby(n,vector<int>());
        for(int u:initial){
            map<int,int> seen;
            dfs(graph,clean,u,seen);
            for(auto x:seen) infetby[x.first].push_back(u);
        }
        vector<int> cont(n,0);
        for(int v=0;v<n;v++){
            int m=infetby[v].size();
            if(m==1) cont[infetby[v][0]]+=1;
        }
        sort(initial.begin(),initial.end());
        int ans=initial[0],anssize=-1;
        for(int u:initial){
            int score=cont[u];
            if(score>anssize||score==anssize&&u<ans){
                ans=u;
                anssize=score;
            }
        }
        return ans;
    }
    void dfs(vector<vector<int>>& graph,vector<int>& clean,int u,map<int,int>& seen){
        for(int i=0;i<graph[u].size();i++){
            if(graph[u][i]==1&&clean[i]==1&&!seen.count(i)){
                seen[i]++;
                dfs(graph,clean,i,seen);
            }
        }
    }
};