310. Minimum Height Trees

最新推荐文章于 2025-05-18 02:45:00 发布

L_Aster

最新推荐文章于 2025-05-18 02:45:00 发布

阅读量151

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/gl486546/article/details/79799134

LeetCode 专栏收录该内容

37 篇文章

订阅专栏

通过从叶节点开始逐步删除的方式寻找图中的最小高度树根节点，采用类似于拓扑排序的方法实现。此算法适用于解决特定类型的图论问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

310. Minimum Height Trees
从叶节点开始，依次删除他们，最后剩下1或者2个节点的就是树根，有点类似拓扑排序的感觉
参考：https://leetcode.com/problems/minimum-height-trees/discuss/76055/Share-some-thoughts

class Solution {
public:
    vector<int> findMinHeightTrees(int n, vector<pair<int, int>>& edges) {
        if(n==1) return {0};
        vector<vector<int>> v(n);
        vector<int> degree(n),leaf;
        for(auto &x:edges){
            v[x.first].push_back(x.second);
            v[x.second].push_back(x.first);
            degree[x.first]++;
            degree[x.second]++;
        }
        for(int i=0;i<n;++i){
            if(degree[i]==1) leaf.push_back(i);
        }
        while(n>2){
            n-=leaf.size();
            vector<int> newleaf;
            for(int w:leaf){
                for(int u:v[w]){
                    degree[u]--;
                    if(degree[u]==1)
                        newleaf.push_back(u);
                }
            }
            leaf=newleaf;
        }
        return leaf;
    }
};