1245. 树的直径(拓扑排序+BFS)

1245. 树的直径

给你这棵「无向树」，请你测算并返回它的「直径」：这棵树上最长简单路径的 边数。

我们用一个由所有「边」组成的数组 edges 来表示一棵无向树，其中 edges[i] = [u, v] 表示节点 u 和 v 之间的双向边。

树上的节点都已经用 {0, 1, ..., edges.length} 中的数做了标记，每个节点上的标记都是独一无二的。

 

示例 1：

输入：edges = [[0,1],[0,2]]
输出：2
解释：
这棵树上最长的路径是 1 - 0 - 2，边数为 2。

示例 2：

输入：edges = [[0,1],[1,2],[2,3],[1,4],[4,5]]
输出：4
解释： 
这棵树上最长的路径是 3 - 2 - 1 - 4 - 5，边数为 4。

 

提示：

• 0 <= edges.length < 10^4
• edges[i][0] != edges[i][1]
• 0 <= edges[i][j] <= edges.length
• edges 会形成一棵无向树

class Solution {
public:
    int treeDiameter(vector<vector<int>>& edges) {
        int n = edges.size() + 1;
        vector<vector<int>> grap(n);
        vector<int> degree(n);
        // 构建无向图
        for(auto &edge : edges){
            grap[edge[0]].push_back(edge[1]);
            grap[edge[1]].push_back(edge[0]);
            degree[edge[0]]++;
            degree[edge[1]]++;
        }

        queue<int> q;
        for (unsigned int i = 0; i < degree.size(); i++) {
            // 将叶子节点(度为1)入队
            if (degree[i] == 1) {
                q.push(i);
            }
        }
        int level = 0; // 处理节点层数
        int count = n; // 剩余节点数
        while (count > 2) { // 剩余节点数小于等于2时找到直径
            int size = q.size();
            for (int i = 0; i < size; i++) {
                int curNode = q.front();
                q.pop();
                degree[curNode]--;
                count--;
                // 剪枝,删除与当前节点相连的边(相连边的对端节点的度--)
                for (auto &peerNode : grap[curNode]) {
                    degree[peerNode]--; // 与当前处理节点相连边对端节点的度--(剪枝)
                    if (degree[peerNode] == 1) { // 叶子节点入队列
                        q.push(peerNode);
                    }
                }
            }
            level++;
        }
        // 剩余节点为2时直径的边数(层数*2)需要再加上1条边(剩余2个节点可以形成一条边)
        return (count % 2 == 0) ? (level * 2 + 1) : level * 2;
    }
};