From ad86fdc99d383323ea14bceb72862dd7cbc6b1f6 Mon Sep 17 00:00:00 2001 From: Jonatan Lazo Date: Sat, 16 May 2020 20:12:43 -0500 Subject: [PATCH] Se implementa la funcio indicada --- src/index.js | 23 ++++++++++++++++++++--- 1 file changed, 20 insertions(+), 3 deletions(-) diff --git a/src/index.js b/src/index.js index 4101a36..471dfab 100644 --- a/src/index.js +++ b/src/index.js @@ -4,7 +4,24 @@ * @returns {number[][]} */ const findNodesWithZeroAndOneParents = (parentChildPairs) => { + let totalParentsMap = new Map(); + parentChildPairs.forEach((element) => { + if (!totalParentsMap.has(element[0])) { + totalParentsMap.set(element[0], 0); + } + let totalParent = totalParentsMap.get(element[1]) || 0; + totalParentsMap.set(element[1], ++totalParent); + }); + let zeroParents = []; + let oneParent = []; + for (let [key, value] of totalParentsMap) { + if (value === 0) { + zeroParents.push(key); + continue; + } + if (value === 1) oneParent.push(key); + } + return [zeroParents, oneParent]; +}; -} - -module.exports = findNodesWithZeroAndOneParents; \ No newline at end of file +module.exports = findNodesWithZeroAndOneParents;