笔试题8 -- 利用拓扑排序解决体育课测验

笔试题8 -- 利用拓扑排序解决体育课测验

原题重现

题目链接:体育课测验(二)_牛客 (nowcoder.com)

体育课共有 numProject 个考核项目,编号为 0 到 numProject−1。考核中每两个项目被划分为一组得到分组数组 groups[i],现规定若想完成项目 groups[i] [0],必须先完成 groups[i] [1]。保证所有分组互不相同,若分组情况能顺利完成考核,请返回任意的一个完成顺序,否则返回空数组。

数据范围:

  1. 1 ≤ numProject ≤ 2000
  2. 1 ≤ groups[i].length ≤ numProject * (numProject−1)

解题思路

通过题目描述 "若想完成项目 groups[i] [0],必须先完成 groups[i] [1]" ,这说明同组的两项目之间存在先后顺序,且每组内刚好有两个项目,他们之间的先后关系可以用箭头来表示 groups[i] [0] <- groups[i] [1] ,所以联想到利用拓扑排序可以有效解决该问题。

示例代码

本代码利用Kahn算法实现拓扑排序

cpp 复制代码
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param numProject int整型 
     * @param groups int整型vector<vector<>> 
     * @return int整型vector
     */
    vector<int> findOrder(int numProject, vector<vector<int> >& groups) {
        vector<vector<int>> edges(numProject);  // 存储边
        vector<int> in(numProject);  // 存储入度

        // 1.建图
        for(auto pair: groups)
        {
            // pair[0] <- pair[1]
            edges[pair[1]].push_back(pair[0]);
            in[pair[0]]++;
        }

        // 2.入度为0的点,加入到队列中
        queue<int> q;
        for(int i = 0; i < numProject; i++)
        {
            if(in[i] == 0) { q.push(i); }
        }

        // 3.拓扑排序(层序遍历)
        vector<int> ret;
        while(!q.empty())
        {
            auto a = q.front();
            q.pop();
            // 塞入结果数组
            ret.push_back(a);
            // 移除该点相关信息
            // a -> b
            for(auto b: edges[a])
            {
                if(--in[b] == 0)  // 已经入度为0的点就可以利用队列准备放入结果数组
                {
                    q.push(b);
                }
            }
        }

        // 4.返回
        if(ret.size() == numProject) { return ret; }
        return {};
    }
};

提交结果:

相关推荐
CoovallyAIHub7 分钟前
一致性模型:单步生成高质量图像,破解扩散模型速度瓶颈
深度学习·算法·计算机视觉
杨筱毅14 分钟前
【穿越Effective C++】条款8:别让异常逃离析构函数——C++异常安全的关键支柱
c++·effective c++
JMzz15 分钟前
Rust 中的数据结构选择与性能影响:从算法复杂度到硬件特性 [特殊字符]
开发语言·数据结构·后端·算法·性能优化·rust
code monkey.32 分钟前
【探寻C++之旅】C++ 智能指针完全指南:从原理到实战,彻底告别内存泄漏
c++·c++11·智能指针
CoovallyAIHub41 分钟前
搞定边缘AI部署:开源神器RamaLama,让视觉语言模型无处不在
深度学习·算法·计算机视觉
CyberSoma42 分钟前
机器人模仿学习运动基元数学编码方法还有用吗?
人工智能·算法·计算机视觉·机器人
CoovallyAIHub1 小时前
英伟达再出「神作」!黄仁勋华盛顿GTC宣布Vera Rubin超级芯片,联手诺基亚进军6G,市值直逼5万亿美元
深度学习·算法·计算机视觉
黑菜钟1 小时前
代码随想录第50天 | 图论 基础介绍(新篇章
算法·深度优先·图论
草莓熊Lotso2 小时前
《算法闯关指南:优选算法--前缀和》--27.寻找数组的中心下标,28.除自身以外数组的乘积
开发语言·c++·算法·rpc
Cx330❀2 小时前
《C++ 继承》三大面向对象编程——继承:派生类构造、多继承、菱形虚拟继承概要
开发语言·c++