返回首页 C++ 模板

树的定义构造模板

邻接表存储树、构建树、遍历树——树形问题的起点

邻接表 建树 GESP6
#include <bits/stdc++.h>
using namespace std;

const int N = 100010;
vector<int> adj[N];  // 邻接表
int parent[N], depth[N];

void buildTree(int n) {
    for (int i = 1; i < n; i++) {
        int u, v;
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
}

// DFS 遍历求深度和父节点
void dfs(int u, int fa) {
    parent[u] = fa;
    depth[u] = depth[fa] + 1;
    for (int v : adj[u]) {
        if (v != fa) dfs(v, u);
    }
}

int main() {
    int n;
    cin >> n;
    buildTree(n);
    dfs(1, 0);  // 以1为根

    for (int i = 1; i <= n; i++)
        cout << i << ": depth=" << depth[i] << " parent=" << parent[i] << endl;
    return 0;
}

📖 要点说明

⚠️ 常见错误