Tree Splitting Hackerrank Solution