【力扣:207,210,310】拓扑排序

cpp 复制代码
class Solution {
    vector<vector<int>>tmp;
    vector<int>index;
public:
    bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
        tmp.resize(numCourses);
        index.resize(numCourses);
        for(auto&i:prerequisites){
            tmp[i[1]].emplace_back(i[0]);
            index[i[0]]++;
        }
        queue<int>que;
        for(int i=0;i<numCourses;i++){
            if(!index[i]) que.emplace(i);
        }
        int count=0;
        while(!que.empty()){
            count++;
            int n=que.front();
            que.pop();
            for(int i:tmp[n]){
                if(--index[i]==0) que.emplace(i);
            }
        }
        return count>=numCourses;
    }
};
cpp 复制代码
class Solution {
    vector<vector<int>>map;
    vector<int>index;
public:
    vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
        map.resize(numCourses);
        index.resize(numCourses);
        for(auto& i:prerequisites){
            map[i[1]].push_back(i[0]);
            index[i[0]]++;
        }
        vector<int>res;
        for(int i=0;i<numCourses;i++) if(!index[i]) res.push_back(i);
        for(int i=0;i<res.size();i++){
            for(int j:map[res[i]]){
                if(!--index[j]){
                    res.push_back(j);
                }
            }
        }
        if(res.size()<numCourses) return {};
        return res;
    }
};

从叶子节点不断向上拓扑,直到剩下至多2个节点作为根节点

cpp 复制代码
class Solution {
public:
    vector<int> findMinHeightTrees(int n, vector<vector<int>>& edges) {
        if(n==1) return {0};
        vector<vector<int>>tree(n);
        vector<int>tmp(n);
        for(auto& i:edges){
            tree[i[0]].push_back(i[1]);
            tree[i[1]].push_back(i[0]);
            tmp[i[0]]++;
            tmp[i[1]]++;
        }
        queue<int>que;
        for(int i=0;i<n;i++){
            if(tmp[i]==1) que.push(i);
        }
        for(int cur=n;cur>2;){
            int size=que.size();
            cur-=size;
            while(size--){
                int fr=que.front();
                que.pop();
                for(int i:tree[fr]){
                    if(--tmp[i]==1) que.push(i);
                }
            }
        }
        vector<int>res;
        while(!que.empty()){
            res.push_back(que.front());
            que.pop();
        } 
        return res;
    }
};
相关推荐
_殊途37 分钟前
《Java HashMap底层原理全解析(源码+性能+面试)》
java·数据结构·算法
还债大湿兄42 分钟前
《C++内存泄漏8大战场:Qt/MFC实战详解 + 面试高频陷阱破解》
c++·qt·mfc
珊瑚里的鱼4 小时前
LeetCode 692题解 | 前K个高频单词
开发语言·c++·算法·leetcode·职场和发展·学习方法
AI+程序员在路上4 小时前
QTextCodec的功能及其在Qt5及Qt6中的演变
开发语言·c++·qt
Risehuxyc4 小时前
C++卸载了会影响电脑正常使用吗?解析C++运行库的作用与卸载后果
开发语言·c++
秋说5 小时前
【PTA数据结构 | C语言版】顺序队列的3个操作
c语言·数据结构·算法
lifallen5 小时前
Kafka 时间轮深度解析:如何O(1)处理定时任务
java·数据结构·分布式·后端·算法·kafka
liupenglove6 小时前
自动驾驶数据仓库:时间片合并算法。
大数据·数据仓库·算法·elasticsearch·自动驾驶
python_tty7 小时前
排序算法(二):插入排序
算法·排序算法
然我7 小时前
面试官:如何判断元素是否出现过?我:三种哈希方法任你选
前端·javascript·算法