并查集
至于并查集的教程,可以参考文章算法学习笔记(1) : 并查集,写的很详细和通俗易懂,本文就不再介绍。下面给出并查集代码:
class UnionFind{
private:
vector<int> parent;
vector<int> rank; // 合并两个不同元素的parent参考依据
public:
UnionFind(int n){
parent = vector<int>(n);
rank = vector<int>(n);
for(int i = 0; i < n ; i++){
parent[i] = i;
}
}
void uni(int x, int y){
int rootx = find(x);
int rooty = find(y);
if(rootx != rooty){
if(rank[rootx] > rank[rooty]){
parent[rooty] = rootx;
}else if(rank[rootx] < rank[rooty]){
parent[rootx] = rooty;
}else{
parent[rooty] = rootx;
rank[rootx]++;
}
}
}
int find(int x){
if(parent[x] != x){
parent[x] = find(parent[x]);
}
return parent[x];
}
};
题目描述
给定一个由不同正整数的组成的非空数组 nums
,考虑下面的图:
- 有
nums.length
个节点,按从nums[0]
到nums[nums.length - 1]
标记; - 只有当
nums[i]
和nums[j]
共用一个大于 1 的公因数时,nums[i]
和nums[j]
之间才有一条边。
返回 图中最大连通组件的大小 。
思路解析
返回图中最大连通组件的大小,换言之,找出图中最大子图的节点数目,可以使用并查集来对其进行求解,对于
nums
中的每一个元素num
,遍历 [2, n u m \sqrt{num} num],找出所有符合公因子的值,其作为合并的桥梁,最终将所有满足要求的元素num
放到同一个集合中去。
说明
- 并查集数组大小为数组
nums
中的最大值。 - 对于范围 [ 2 , n u m ] [2, \sqrt{num}] [2,num] 内的每个正整数 i i i,如果 i i i 是 n u m num num 的因数,则 n u m num num 和 i i i、 n u m i \frac{num}{i} inum 都属于同一个组件。
完整代码
#include <iostream>
#include <vector>
#include <bits/stdc++.h>
using namespace std;
class UnionFind{
private:
vector<int> parent;
vector<int> rank;
public:
UnionFind(int n){
parent = vector<int>(n);
rank = vector<int>(n);
for(int i = 0; i < n ; i++){
parent[i] = i;
}
}
void uni(int x, int y){
int rootx = find(x);
int rooty = find(y);
if(rootx != rooty){
if(rank[rootx] > rank[rooty]){
parent[rooty] = rootx;
}else if(rank[rootx] < rank[rooty]){
parent[rootx] = rooty;
}else{
parent[rooty] = rootx;
rank[rootx]++;
}
}
}
int find(int x){
if(parent[x] != x){
parent[x] = find(parent[x]);
}
return parent[x];
}
};
class Solution {
public:
int largestComponentSize(vector<int>& nums) {
int m = *max_element(nums.begin(), nums.end()); // 找到vector的最大元素
UnionFind uf(m + 1);
for(int num : nums){
for(int i = 2; i * i <= num; i++){
if( num % i == 0){
uf.uni(num, i);
uf.uni(num, num / i);
}
}
}
vector<int> counts(m + 1);
int ans = 0;
for(int num : nums){
int root = uf.find(num);
counts[root]++;
ans = max(ans, counts[root]);
}
return ans;
}
};
int main(){
// vector<int> nums = {4, 6, 15, 35};
// vector<int> nums = {20, 50, 9, 63};
vector<int> nums = {2, 3, 6, 7, 4, 12, 21, 39};
Solution so;
int ans = so.largestComponentSize(nums);
cout<< ans << endl;
system("pause");
return 0;
}
// output: 8