码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode.310最小高度树

时间:2020-01-05 15:20:39      阅读:80      评论:0      收藏:0      [点我收藏+]

标签:著作权   假设   一个   返回   深度   div   来源   不能   imu   

对于一个具有树特征的无向图,我们可选择任何一个节点作为根。图因此可以成为树,在所有可能的树中,具有最小高度的树被称为最小高度树。给出这样的一个图,写出一个函数找到所有的最小高度树并返回他们的根节点。

格式

该图包含 n 个节点,标记为 0 到 n - 1。给定数字 n 和一个无向边 edges 列表(每一个边都是一对标签)。

你可以假设没有重复的边会出现在 edges 中。由于所有的边都是无向边, [0, 1]和 [1, 0] 是相同的,因此不会同时出现在 edges 里。

示例 1:

输入: n = 4, edges = [[1, 0], [1, 2], [1, 3]]

0
|
1
/ \
2 3

输出: [1]
示例 2:

输入: n = 6, edges = [[0, 3], [1, 3], [2, 3], [4, 3], [5, 4]]

0 1 2
\ | /
3
|
4
|
5

输出: [3, 4]
说明:

 根据树的定义,树是一个无向图,其中任何两个顶点只通过一条路径连接。 换句话说,一个任何没有简单环路的连通图都是一棵树。
树的高度是指根节点和叶子节点之间最长向下路径上边的数量。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-height-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

 

第一次尝试是用深度优先算法去计算每个点作为根时的高度,然后再找出最小值,结果时间复杂度太大。(我也想不清楚有多大)

第二次发现可以多次去除叶子结点,即度数为1的点。当一次去除之后发现所有的点都被去除时,这次去除的点就是答案。不过在每次去除时要注意不能同时判断同时去除,那样会导致次叶子结点在同一轮被删除。应当先记录哪些点需要去除再统一的进行去除和删边。

class Solution {
public:
    int visited[100000],deg[100000];
/*    int dfs(int t,int heigh, vector<vector<int>>& edges){
        int max=-1,o,i;
        for(i=0;i<edges.size();i++){
            if(edges[i][0]==t&&visited[edges[i][1]]==0){
                visited[edges[i][1]]=1;
                o=dfs(edges[i][1],heigh+1,edges);
                if(o>max) max=o;
                visited[edges[i][1]]=0;
            }else if(edges[i][1]==t&&visited[edges[i][0]]==0){
                visited[edges[i][0]]=1;
                o=dfs(edges[i][0],heigh+1,edges);
                if(o>max) max=o;
                visited[edges[i][0]]=0;
            }
        }
        if(max==-1){
            return heigh;
        }
        return max;
    }
    int height(int k,int n, vector<vector<int>>& edges){

        for(i=0;i<n;i++){
            visited[i]=0;
        }

    }*/
    /*vector<int> findMinHeightTrees(int n, vector<vector<int>>& edges) {
        int i,ii;
        for(i=0;i<n;i++){
            visited[i]=0;
        }
        vector<int> res;
        int r,min=10000000;
        for(i=0;i<n;i++){
            visited[i]=1;
            r=dfs(i,0,edges);
            if(r<min){
                res.clear();
                res.push_back(i);
                min=r;
            }else if(r==min){
                res.push_back(i);
            }
            //cout<<i<<"  "<<r<<endl;
            visited[i]=0;
        }
        return res;
    }*/
        vector<int> findMinHeightTrees(int n, vector<vector<int>>& edges) {
            int i;
            int live[n];
            for(i=0;i<n;i++){
                deg[i]=0;
                live[i]=1;
            }
            for(i=0;i<edges.size();i++){
                deg[edges[i][0]]++;
                deg[edges[i][1]]++;
            }
            vector<int> res;
            int t=1,j,z=n;
            while(z>0){
                res.clear();
                for(i=0;i<n;i++){
                    if(deg[i]<=t&&live[i]==1){
                        res.push_back(i);
                        }
                    }
                        for(i=0;i<res.size();i++){
                        for(j=0;j<edges.size();j++){
                            if(edges[j][0]==res[i]&&live[edges[j][1]]==1){
                                deg[edges[j][1]]--;
                            }
                            if(edges[j][1]==res[i]&&live[edges[j][0]]==1){
                                deg[edges[j][0]]--;
                            }
                            }
                    z--;
                    live[res[i]]=0;
                    //cout<<"t="<<t<<"  "<<i<<" is moved\n";
                        }
                }
            return res;
        }
};

leetcode.310最小高度树

标签:著作权   假设   一个   返回   深度   div   来源   不能   imu   

原文地址:https://www.cnblogs.com/hyffff/p/12152353.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!