【图论】树的直径

树的直径即为一棵树中距离最远的两点之间的路径

方法一:DFS

先以任意一点为起点跑一遍dfs,记录离起点距离最远的点p(这个点一定是直径的一个端点,感性理解一下不证明了),然后再以最远点再跑一遍dfs,记录此时距离最远的点q,那么pq就是该树的直接

树中有负权边时不可以用这个方法

cpp 复制代码
const int N = 10000 + 10;

int n, c, d[N];
vector<int> g[N];

void dfs(int u, int fa)
{
    for (int v : E[u])
    {
        if (v == fa) continue;
        d[v] = d[u] + 1; // 如边有权值,把1换成权值即可
        if (d[v] > d[c]) c = v; // 更新最大距离的点
        dfs(v, u);
    }
}

int main()
{
    cin >> n;
    for (int i = 1; i < n; i++)
    {
        int u, v;
        scanf("%d %d", &u, &v);
        g[u].push_back(v), g[v].push_back(u);
    }
    dfs(1, 0); // 第一遍dfs
    int p = c; // 一个端点
    d[c] = 0;
    dfs(c, 0); // 第二遍dfs
    int q = c; // 另一个端点
    cout << d[c];
    return 0;
}

方法二:树形dp

dp[u]为以u为根的子树中离u最远的点的路径长度

转移方程(v为u的子结点):dp[u] = max(dp[u], dp[v] + w(u, v))

两条经过根结点的最长路径即为该子树中的直径

转移方程:zj = max(zj, dp[u] + dp[v] + w(u, v))

cpp 复制代码
const int N = 10000 + 10;

int n, zj = 0;
int dp[N];
vector<int> g[N];

void dfs(int u, int fa)
{
    for (int v : E[u])
    {
        if (v == fa) continue;
        dfs(v, u);
        zj = max(zj, dp[u] + dp[v] + 1); // 如为有权边,把1换成权值即可
        dp[u] = max(dp[u], dp[v] + 1); // 如为有权边,把1换成权值即可
    }
}

int main()
{
    cin >> n;
    for (int i = 1; i < n; i++)
    {
        int u, v;
        cin >> u >> v;
        g[u].push_back(v), g[v].push_back(u);
    }
    dfs(1, 0);
    cout << zj << '\n';
    return 0;
}
相关推荐
java1234_小锋4 分钟前
Scikit-learn Python机器学习 - 分类算法 - K-近邻(KNN)算法
python·算法·机器学习
智者知已应修善业7 分钟前
【矩阵找最大小所在位置】2022-11-13
c语言·c++·经验分享·笔记·算法·矩阵
shan&cen11 分钟前
Day04 前缀和&差分 1109. 航班预订统计 、304. 二维区域和检索 - 矩阵不可变
java·数据结构·算法
手握风云-16 分钟前
回溯剪枝的 “减法艺术”:化解超时危机的 “救命稻草”(二)
算法·机器学习·剪枝
QiZhang | UESTC29 分钟前
JAVA算法练习题day11
java·开发语言·python·算法·hot100
屁股割了还要学33 分钟前
【数据结构入门】排序算法(4)归并排序
c语言·数据结构·学习·算法·排序算法
努力学习的小廉1 小时前
我爱学算法之—— 位运算(上)
c++·算法
ゞ 正在缓冲99%…1 小时前
leetcode35.搜索插入位置
java·算法·leetcode·二分查找
lifallen1 小时前
字节跳动Redis变种Abase:无主多写架构如何解决高可用难题
数据结构·redis·分布式·算法·缓存
feifeigo1232 小时前
星座SAR动目标检测(GMTI)
人工智能·算法·目标跟踪