diff --git a/src/index.js b/src/index.js index 4101a36..a7e0d28 100644 --- a/src/index.js +++ b/src/index.js @@ -4,7 +4,17 @@ * @returns {number[][]} */ const findNodesWithZeroAndOneParents = (parentChildPairs) => { - + let uniqueItems = parentChildPairs.flat(); + uniqueItems = uniqueItems.filter((item, index) => uniqueItems.indexOf(item) == index) + const parentsCount = new Array(uniqueItems.length).fill(0); + parentChildPairs.map(pairs =>{ + parentsCount[uniqueItems.indexOf(pairs[1])]++; + }); + const result = [[],[]]; + parentsCount.map((count, index) => { + if(count < 2) result[count].push(uniqueItems[index]) + }); + return result; } module.exports = findNodesWithZeroAndOneParents; \ No newline at end of file