tree_zig_zag - working on C++ solution