《算法导论》原书第3版介绍图的两种表达方法,邻接链表和邻接矩阵;邻接链表由V个链表构成的数组;但是链表形式无法在常量时间复杂度查询两个结点是否相邻。用unordered_set代替链表,用unordered_map代替数组,实现常量时间查询两个结点是否相邻。
#include <iostream>
#include <unordered_map>
#include <unordered_set>
using std::cout;
using std::endl;
template <typename T>
class Graph {
public:
Graph() {}
void Add(const T& a, const T& b) { data_[a].insert(b); }
bool IsLinked(const T& a, const T& b) {
auto ia = data_.find(a);
if (ia != data_.end()) {
auto ib = ia->second.find(b);
if (ib != ia->second.end()) {
return true;
}
}
return false;
}
void Remove(const T& a, const T& b) {
if (IsLinked(a, b)) {
data_[a].erase(b);
}
}
private:
std::unordered_map<T, std::unordered_set<T>> data_;
};
int main() {
Graph<int32_t> ag;
int32_t v = 299792258;
ag.Add(0, v);
cout << ag.IsLinked(1, 3) << " " << ag.IsLinked(0, v) << endl;
ag.Remove(0, v);
cout << ag.IsLinked(1, 3) << " " << ag.IsLinked(0, v) << endl;
return 0;
}