1 Star 0 Fork 0

yuhang2__2/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
graph-valid-tree.cpp 1.99 KB
一键复制 编辑 原始数据 按行查看 历史
kamyu 提交于 6年前 . Update graph-valid-tree.cpp
// Time: O(|V| + |E|)
// Space: O(|V| + |E|)
// Same complexity, but faster version.
class Solution {
public:
bool validTree(int n, vector<pair<int, int>>& edges) {
if (edges.size() != n - 1) {
return false;
}
unordered_map<int, vector<int>> neighbors;
for (const auto& edge : edges) {
neighbors[edge.first].emplace_back(edge.second);
neighbors[edge.second].emplace_back(edge.first);
}
queue<int> q;
q.emplace(0);
unordered_set<int> visited;
visited.emplace(0);
while (!q.empty()) {
const int i = q.front();
q.pop();
for (const auto& node : neighbors[i]) {
if (!visited.count(node)) {
visited.emplace(node);
q.emplace(node);
}
}
}
return visited.size() == n;
}
};
// Time: O(|V| + |E|)
// Space: O(|V| + |E|)
class Solution2 {
public:
struct node {
int parent = -1;
vector<int>neighbors;
};
bool validTree(int n, vector<pair<int, int>>& edges) {
unordered_map<int, node> nodes;
for (const auto& edge : edges) {
nodes[edge.first].neighbors.emplace_back(edge.second);
nodes[edge.second].neighbors.emplace_back(edge.first);
}
queue<int> q;
q.emplace(0);
unordered_set<int> visited;
visited.emplace(0);
while (!q.empty()) {
const int i = q.front();
q.pop();
for (const auto& node : nodes[i].neighbors) {
if (node != nodes[i].parent) {
if (visited.find(node) != visited.end()) {
return false;
} else {
visited.emplace(node);
nodes[node].parent = i;
q.emplace(node);
}
}
}
}
return visited.size() == n;
}
};
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

搜索帮助