LeetCode.924 儘量減少惡意軟體的傳播 Minimize Malware Spread(並查集)
阿新 • • 發佈:2019-02-11
在節點網路中,只有當 graph[i][j] = 1 時,每個節點 i 能夠直接連線到另一個節點 j。
一些節點 initial 最初被惡意軟體感染。只要兩個節點直接連線,且其中至少一個節點受到惡意軟體的感染,那麼兩個節點都將被惡意軟體感染。這種惡意軟體的傳播將繼續,直到沒有更多的節點可以被這種方式感染。
假設 M(initial) 是在惡意軟體停止傳播之後,整個網路中感染惡意軟體的最終節點數。
我們可以從初始列表中刪除一個節點。如果移除這一節點將最小化 M(initial), 則返回該節點。如果有多個節點滿足條件,就返回索引最小的節點。
請注意,如果某個節點已從受感染節點的列表 initial 中刪除,它以後可能仍然因惡意軟體傳播而受到感染。
示例 1:
輸入:graph = [[1,1,0],[1,1,0],[0,0,1]], initial = [0,1]
輸出:0
示例 2:
輸入:graph = [[1,0,0],[0,1,0],[0,0,1]], initial = [0,2]
輸出:0
示例 3:
輸入:graph = [[1,1,1],[1,1,1],[1,1,1]], initial = [1,2]
輸出:1
提示:
1 < graph.length = graph[0].length <= 300 0 <= graph[i][j] == graph[j][i] <= 1 graph[i][i] = 1 1 <= initial.length < graph.length 0 <= initial[i] < graph.length
題目很簡單,很明顯的並查集應用。
把各個邊的兩個節點放到並查集當中,處理完之後,對每個initial當中的節點進行計數,如果當前節點的連通的節點中只有他一個病變節點,統計一下這個集合節點數量。然後保留最大的那個即可。
class Solution{
private:
int father[305];
int node_counter[305];
int source_counter[305];
int find(int p)
{
if(p==father[p])
return p;
else
{
father[p]=find(father[p]);
return father[p];
}
}
void make_union(int p,int q)
{
int x,y;
x=find(p);
y=find(q);
if(x!=y)
father[x]=y;
}
public:
int minMalwareSpread(vector<vector<int>>& graph,vector<int>& initial)
{
int minus_node=0,n=graph.size();
sort(initial.begin(),initial.end());
int result=initial[0];
for(int i=0;i<n;++i)
{
father[i]=i;
node_counter[i]=0;
source_counter[i]=0;
}
for(int i=0;i<n;++i)
{
for(int j=0;j<i;++j)
{
if(graph[i][j])
make_union(i,j);
}
}
for(int i=0;i<n;++i)
node_counter[find(father[i])]+=1;
for(int i=0;i<initial.size();++i)
source_counter[find(father[initial[i]])]+=1;
for(int i=0;i<initial.size();++i)
{
if(source_counter[find(father[initial[i]])]==1 && node_counter[find(father[initial[i]])]>minus_node)
{
minus_node=node_counter[find(father[initial[i]])];
result=initial[i];
}
}
return result;
}
};