With given output sample, it seems that this problem is actually based on the assumption that the root is 1 - and the edge is actually directed but never rely on the input, is that right?