#include <iostream>#include <vector>using namespace std;struct Edge { int u, v;};int n, m;vector<Edge> e;vector<bool> vis;bool find_edge(int u, int v) { for (int i = 1; i <= m; ++i) { if (e[i].u == u && e[i].v == v) { return true; } } return false;}void dfs(int u) { if (vis[u]) return; vis[u] = true; for (int i = 1; i <= m; ++i) { if (e[i].u == u) { dfs(e[i].v); } }}int main() { cin >> n >> m; vis.resize(n + 1, false); e.resize(m + 1); for (int i = 1; i <= m; ++i) cin >> e[i].u >> e[i].v; return 0;}
Python
class Edge: def __init__(self, u=0, v=0): self.u = u self.v = vn, m = map(int, input().split())e = [Edge() for _ in range(m)]vis = [False] * nfor i in range(m): e[i].u, e[i].v = map(int, input().split())def find_edge(u, v): for i in range(m): if e[i].u == u and e[i].v == v: return True return Falsedef dfs(u): if vis[u]: return vis[u] = True for i in range(m): if e[i].u == u: dfs(e[i].v)
使用一个二维数组 adj 来存边,其中 adj[u][v] 为 1 表示存在 u 到 v 的边,为 0 表示不存在。如果是带边权的图,可以在 adj[u][v] 中存储 u 到 v 的边的边权。
参考代码
[list2tab]
C++
#include <iostream>#include <vector>using namespace std;int n, m;vector<bool> vis;vector<vector<bool>> adj;bool find_edge(int u, int v) { return adj[u][v]; }void dfs(int u) { if (vis[u]) return; vis[u] = true; for (int v = 1; v <= n; ++v) { if (adj[u][v]) { dfs(v); } }}int main() { cin >> n >> m; vis.resize(n + 1); adj.resize(n + 1, vector<bool>(n + 1)); for (int i = 1; i <= m; ++i) { int u, v; cin >> u >> v; adj[u][v] = true; } return 0;}
Python
vis = [False] * (n + 1)adj = [[False] * (n + 1) for _ in range(n + 1)]for i in range(1, m + 1): u, v = map(lambda x: int(x), input().split()) adj[u][v] = Truedef find_edge(u, v): return adj[u][v]def dfs(u): if vis[u]: return vis[u] = True for v in range(1, n + 1): if adj[u][v]: dfs(v)
使用一个支持动态增加元素的数据结构构成的数组,如 vector<int> adj[n + 1] 来存边,其中 adj[u] 存储的是点 u 的所有出边的相关信息(终点、边权等)。
参考代码
[list2tab]
C++
#include <iostream>#include <vector>using namespace std;int n, m;vector<bool> vis;vector<vector<int>> adj;bool find_edge(int u, int v) { for (int i = 0; i < adj[u].size(); ++i) { if (adj[u][i] == v) { return true; } } return false;}void dfs(int u) { if (vis[u]) return; vis[u] = true; for (int i = 0; i < adj[u].size(); ++i) dfs(adj[u][i]);}int main() { cin >> n >> m; vis.resize(n + 1); adj.resize(n + 1); for (int i = 1; i <= m; ++i) { int u, v; cin >> u >> v; adj[u].push_back(v); } return 0;}
Python
vis = [False] * (n + 1)adj = [[] for _ in range(n + 1)]for i in range(1, m + 1): u, v = map(lambda x: int(x), input().split()) adj[u].append(v)def find_edge(u, v): for i in range(0, len(adj[u])): if adj[u][i] == v: return True return Falsedef dfs(u): if vis[u]: return vis[u] = True for i in range(0, len(adj[u])): dfs(adj[u][i])
复杂度
查询是否存在 u 到 v 的边:O(d+(u))(如果事先进行了排序就可以使用 二分查找 做到 O(log(d+(u))))。
// head[u] 和 cnt 的初始值都为 -1void add(int u, int v) { nxt[++cnt] = head[u]; // 当前边的后继(指向旧的第一条边) head[u] = cnt; // 更新起点 u 的第一条边为当前边 to[cnt] = v; // 当前边的终点}// 遍历 u 的出边for (int i = head[u]; ~i; i = nxt[i]) { // ~i 表示 i != -1 int v = to[i]; // v 是边 i 的终点 // 处理边 u->v (边编号为 i)}
Python
# head[u] 和 cnt 的初始值都为 -1def add(u, v): cnt = cnt + 1 nex[cnt] = head[u] # 当前边的后继 head[u] = cnt # 起点 u 的第一条边 to[cnt] = v # 当前边的终点# 遍历 u 的出边i = head[u]while ~i: # ~i 表示 i != -1 v = to[i] i = nxt[i]
参考代码
#include <iostream>#include <vector>using namespace std;int n, m;vector<bool> vis;vector<int> head, nxt, to;void add(int u, int v) { nxt.push_back(head[u]); head[u] = to.size(); to.push_back(v);}bool find_edge(int u, int v) { for (int i = head[u]; ~i; i = nxt[i]) { // ~i 表示 i != -1 if (to[i] == v) { return true; } } return false;}void dfs(int u) { if (vis[u]) return; vis[u] = true; for (int i = head[u]; ~i; i = nxt[i]) dfs(to[i]);}int main() { cin >> n >> m; vis.resize(n + 1, false); head.resize(n + 1, -1); for (int i = 1; i <= m; ++i) { int u, v; cin >> u >> v; add(u, v); } return 0;}
复杂度
查询是否存在 u 到 v 的边:O(d+(u))。
遍历点 u 的所有出边:O(d+(u))。
遍历整张图:O(n+m)。
空间复杂度:O(m)。
应用
存各种图都很适合,但不能快速查询一条边是否存在,也不能方便地对一个点的出边进行排序。
优点
边有编号: 每条边都有唯一的编号 cnt,这在网络流等算法中很有用
反向边技巧: 如果 cnt 的初始值为奇数,存双向边时 i ^ 1 即是 i 的反边(常用于 网络流)
struct Edge { int to, w, nxt; // 终点、权值、下一条边};Edge edge[MAXM];int head[MAXN], cnt = -1;void add(int u, int v, int w) { edge[++cnt] = {v, w, head[u]}; head[u] = cnt;}// 遍历for (int i = head[u]; ~i; i = edge[i].nxt) { int v = edge[i].to; int w = edge[i].w;}