【算法】【华为机试】 神经网络的最短执行时间:给出N个节点,每个节点的信息包含该节点执行时间、下一个节点列表,求神经网络的最短执行时间。

本文介绍了一种计算神经网络中最短执行时间的方法,通过构建有向无环图并使用拓扑排序来找出最长路径,从而确定整个网络的最小执行时间。

【编程 | 100】 神经网络的最短执行时间

题目描述
给出N个节点,每个节点的信息包含该节点执行时间、下一个节点列表,求神经网络的最短执行时间。
假设深度学习模型是一个有向无环图。若算子A依赖算子B的输出,则当B执行完后才能计算A,如果没有依赖关系,则可并行执行,
计算每个网络所需要的最短时间。注意(算子索引从0开始
有向无环图求拓扑排序。

示例
输入

7
A 10 1 2 3
B 9 3 4 5 
C 22 4 5
D 20
E 19
F 18 6
G 21

输出

71

A->C->F->G

其实是 最长路径 问题

  1. 从A出发,找其前序节点,发现无前序节点,记录A的前序节点为空,当前最大路径为10,记录 Array[A]=10
  2. 从B出发,找前序节点,找到A,前序节点只有A,记录前序节点为A,当前最大路径为9+Array[A] = 19,记录Array[B] = 19
  3. 从C出发,找前序节点,找到A,同理,记录Array[C]=22+Array[A]=32
  4. 从D出发,找到前序节点,找到了前序节点A和B,Array[A]=10,Array[B]=19所以选最大的,选择前序节点B。
  5. 从E出发,找到前序节点是B或者C,Array[B]<Array[c],选择C,得到32+Value[E] = 32+19=51
  6. 从F出发,同理找到前序节点,并选择C,得到 18+32=50
  7. 从G出发,前序节点是F,得到21+50=71

代码:

#include <iostream>

using namespace std;

void FindFront(int* nodes, int rows, int cols, int* fronts,int aim) {
   
   
	for (int i = 0; i < rows; i++)
	{
   
   
		for (int j = 0; j < cols; j++) {
   
   
			if (nodes[i*rows+j]==aim)
			{
   
   
				fronts[i] = 1; 
			}
			else if (nodes[i * rows + j] == -1)
			{
   
   
				break;
			}
		}
	}
}

struct nodeInfo {
   
   
	int index;
	int sum;
};
int MinActingTime(int* nodes, int rows, int cols) {
   
   
	if (nodes==nullptr||rows<=0||cols<=0)
	{
   
   
		return 0; // 处理异常情况
	}
	nodeInfo* maxArray = new nodeInfo[rows];
	for (int i = 0; i < rows; i++)
	{
   
   
		maxArray[i].index = -1;
		maxArray[i]
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值