华为OD机试 - 查找一个有向网络的头节点和尾节点(Java & JS & Python & C & C++)

在有向图中,头节点是入度为0的节点,尾节点是出度为0的节点。

以下是针对不同编程语言的解决方案:

  1. Java:



import java.util.*;
 
public class Solution {
    /**
     * 获取头节点和尾节点
     * @param graph 图的邻接表表示
     * @return 头节点和尾节点数组
     */
    public int[] getHeadAndTailNodes(List<List<Integer>> graph) {
        int[] degrees = new int[graph.size()];
        boolean[] isHead = new boolean[graph.size()];
        boolean[] isTail = new boolean[graph.size()];
        
        // 计算每个节点的度
        for (int i = 0; i < graph.size(); i++) {
            for (int neighbor : graph.get(i)) {
                degrees[neighbor]++;
            }
        }
        
        // 标记头节点和尾节点
        for (int i = 0; i < graph.size(); i++) {
            if (degrees[i] == 0) {
                isHead[i] = true;
            }
            if (graph.get(i).isEmpty()) {
                isTail[i] = true;
            }
        }
        
        // 统计头节点和尾节点的数量
        int headCount = 0, tailCount = 0;
        for (int i = 0; i < isHead.length; i++) {
            if (isHead[i]) {
                headCount++;
            }
            if (isTail[i]) {
                tailCount++;
            }
        }
        
        // 不存在头节点或尾节点的情况
        if (headCount == 0 || tailCount == 0) {
            return new int[]{-1, -1};
        }
        
        // 寻找唯一的头节点和尾节点
        int headNode = -1, tailNode = -1;
        for (int i = 0; i < isHead.length; i++) {
            if (isHead[i]) {
                if (headNode == -1) {
                    headNode = i;
                } else {
                    headNode = -1; // 不唯一
                    break;
                }
            }
            if (isTail[i]) {
                if (tailNode == -1) {
                    tailNode = i;
                } else {
                    tailNode = -1; // 不唯一
                    break;
                }
            }
        }
        
        return new int[]{headNode, tailNode};
    }
}
  1. JavaScript:



/**
 * @param {number[][]} graph
 * @return {number[]}
 */
var getHeadAndTailNodes = function(graph) {
    let degrees = new Array(graph.length).fill(0);
    let isHead = new Array(graph.length).fill(false);
    let isTail = new Array(graph.length).fill(false);
    
    // 计算每个节点的度
    for (let i = 0; i < graph.length; i++) {
        for (let neighbor of graph[i]) {
            degrees[neighbor]++;
        }
    }
    
    //

评论已关闭

推荐阅读

Vue中使用mind-map实现在线思维导图
2024年08月04日
VUE
Web前端最全Vue实现免密登录跳转的方式_vue怎么样不登录返回首页,最强技术实现
2024年08月04日
VUE
vue3 项目搭建教程(基于create-vue,vite,Vite + Vue)
2024年08月04日
VUE
Vue-颜色选择器实现方案——>Vue-Color( 实战*1+ Demo*7)
2024年08月04日
VUE
Vue项目卡顿慢加载?这些优化技巧告诉你!_vue数据多渲染卡顿
2024年08月04日
VUE
vue中的keep-alive详解与应用场景
2024年08月04日
VUE
Vue、React实现excel导出功能(三种实现方式保姆级讲解)
2024年08月04日
vue-office/docx插件实现docx文件预览
2024年08月04日
VUE
java调用js文件的两种方法(支持V8引擎)
2024年08月04日
JavaScript:解决计算精度问题/mathjs/bignumber.js/big.js/decimal.js
2024年08月04日
两周从爬虫小白变大神 _yjs_js_security_passport
2024年08月04日
JS笔记(对象、函数、数组)
2024年08月04日
Markdown.js:强大的纯JavaScript Markdown解析器
2024年08月04日
Vue项目:js模拟点击a标签下载文件并重命名,URL文件地址下载方法、请求接口下载文件方法总结。
2024年08月04日
vue 父组件怎么获取子组件里面的data数据
2024年08月04日
VUE
个人开发实现AI套壳网站快速搭建(Vue+elementUI+SpringBoot)
2024年08月04日
el-table 表格封装并改造实现单元格可编辑
2024年08月04日
none
nodejs环境下创建vue项目、SSH密钥登陆!!!
2024年08月04日
vue+quill+element-ui实现视频、图片上传及缩放保姆级教程,轻松使用富文本
2024年08月04日
【three.js】22. Imported Models导入模型
2024年08月04日