Data structure (Python)

The correct answer for hw4 should be the same in the picture.

And total nodes should not be larger than 10000 nodes.