Python. Fail to build tree base on the result of depth first search(with height of each node given). Not sure what's the problem when backtracking

Question

The data input is a list with the hierarchy.

[node0 : 0(height), node1: 1 , node2: 2, node3:2, node4:1, node5: 2]

The tree I want then would