1 Star 0 Fork 0

yuhang2__2/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
graph-connectivity-with-threshold.cpp 1.76 KB
一键复制 编辑 原始数据 按行查看 历史
// Time: O((nlogn + q) * α(n)) ~= O(nlogn + q)
// Space: O(n)
class Solution {
public:
vector<bool> areConnected(int n, int threshold, vector<vector<int>>& queries) {
UnionFind union_find(n);
for (int i = threshold + 1; i <= n; ++i) {
// https://stackoverflow.com/questions/25905118/finding-big-o-of-the-harmonic-series
// sum of harmonic series is O(logn)
for (int j = 2 * i; j <= n; j += i) {
union_find.union_set(i - 1, j - 1);
}
}
vector<bool> result;
for (const auto& q : queries) {
result.emplace_back(union_find.find_set(q[0] - 1) == union_find.find_set(q[1] - 1));
}
return result;
}
private:
class UnionFind {
public:
UnionFind(const int n) : set_(n), rank_(n), count_(n) {
iota(set_.begin(), set_.end(), 0);
}
int find_set(const int x) {
if (set_[x] != x) {
set_[x] = find_set(set_[x]); // Path compression.
}
return set_[x];
}
bool union_set(const int x, const int y) {
int x_root = find_set(x), y_root = find_set(y);
if (x_root == y_root) {
return false;
}
if (rank_[x_root] < rank_[y_root]) { // Union by rank.
set_[x_root] = y_root;
} else if (rank_[x_root] > rank_[y_root]) {
set_[y_root] = x_root;
} else {
set_[y_root] = x_root;
++rank_[x_root];
}
--count_;
return true;
}
int size() const {
return count_;
}
private:
vector<int> set_;
vector<int> rank_;
int count_;
};
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/yuhang2__2/LeetCode-Solutions.git
git@gitee.com:yuhang2__2/LeetCode-Solutions.git
yuhang2__2
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助