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

warning: 这篇文章距离上次修改已过186天,其中的内容可能已经有所变动。

在有向图中,头节点是入度为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]++;
        }
    }
    
    //

评论已关闭

推荐阅读

DDPG 模型解析,附Pytorch完整代码
2024年11月24日
DQN 模型解析,附Pytorch完整代码
2024年11月24日
AIGC实战——Transformer模型
2024年12月01日
Socket TCP 和 UDP 编程基础(Python)
2024年11月30日
python , tcp , udp
如何使用 ChatGPT 进行学术润色?你需要这些指令
2024年12月01日
AI
最新 Python 调用 OpenAi 详细教程实现问答、图像合成、图像理解、语音合成、语音识别(详细教程)
2024年11月24日
ChatGPT 和 DALL·E 2 配合生成故事绘本
2024年12月01日
omegaconf,一个超强的 Python 库!
2024年11月24日
【视觉AIGC识别】误差特征、人脸伪造检测、其他类型假图检测
2024年12月01日
[超级详细]如何在深度学习训练模型过程中使用 GPU 加速
2024年11月29日
Python 物理引擎pymunk最完整教程
2024年11月27日
MediaPipe 人体姿态与手指关键点检测教程
2024年11月27日
深入了解 Taipy:Python 打造 Web 应用的全面教程
2024年11月26日
基于Transformer的时间序列预测模型
2024年11月25日
Python在金融大数据分析中的AI应用(股价分析、量化交易)实战
2024年11月25日
AIGC Gradio系列学习教程之Components
2024年12月01日
Python3 `asyncio` — 异步 I/O,事件循环和并发工具
2024年11月30日
llama-factory SFT系列教程:大模型在自定义数据集 LoRA 训练与部署
2024年12月01日
Python 多线程和多进程用法
2024年11月24日
Python socket详解,全网最全教程
2024年11月27日