LeetCode 310:最小高度树

最新推荐文章于 2024-03-21 09:30:55 发布

MusicLEEE

最新推荐文章于 2024-03-21 09:30:55 发布

阅读量468

点赞数

分类专栏： Leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/qq_39731130/article/details/123996931

版权

Leetcode 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

LeetCode 310:最小高度树

原题地址

基本思路

从任意一个顶点对图进行BFS，找到离原点最远的点A；
再从点A对图进行BFS，找到距离A最远的点B。
使用DFS找到AB两点之间的路径。
路径的中点，即为最小高度树的根。

错误思路

使用了暴力BFS方法，即对每个顶点进行BFS，计算以各个顶点为根的树高度。这样时间复杂度为O(n²)，无法通过所有测试。

代码

class Solution {
public:

    int bfs(std::vector<std::vector<int>>& nodeList, int root)
    {
        std::vector<bool> visit(nodeList.size(), false);
        std::vector<int> height(nodeList.size(), -1);
        std::queue<int> q;
        q.push(root);
        visit[root]=true;
        height[root]=0;

        int maxLength = 0;
        int maxTarget = root;
        while(!q.empty())
        {
            int head = q.front();
            q.pop();
            for(int i=0;i<nodeList[head].size();i++)
            {
                int target = nodeList[head][i];
                if(!visit[target])
                {
                    visit[target]=true;
                    height[target]=height[head]+1;
                    q.push(target);
                    if(height[target] >  maxLength)
                    {
                        maxLength=height[target];
                        maxTarget=target;
                    }
                }
            }
        }
        return maxTarget;
    }

    bool dfs(std::vector<std::vector<int>>& nodeList, std::vector<int> &path,
     std::vector<bool>& visit, int root, int second)
    {
        if(visit[root]) return false;

        visit[root]=true;
        path.push_back(root);
        if(root == second)
        {
            return true;
        }
        bool isPath = false;
        for(int i=0;i<nodeList[root].size();i++)
        {
            if(dfs(nodeList, path, visit, nodeList[root][i], second))
            {
                isPath=true;
                break;
            }
        }
        if(isPath)
        {
            return true;
        }
        else
        {
            path.pop_back();
            return false;
        }
    }

    vector<int> findMinHeightTrees(int n, vector<vector<int>>& edges) {
        
        std::vector<std::vector<int>>  nodeList(n);
        for(int i=0;i<edges.size();i++)
        {
            int head = edges[i][0];
            int rear = edges[i][1];
            nodeList[head].push_back(rear);
            nodeList[rear].push_back(head);
        }
        int first = bfs(nodeList,0);
        int second = bfs(nodeList,first);
        std::vector<int> path;
        std::vector<bool> visit(n, false);
        dfs(nodeList,path,visit,first,second);

        if(path.size()%2 == 1)  
            return {path[path.size()/2]};
        else 
            return {path[path.size()/2-1], path[path.size()/2]};
    }
};