In general, I don't like adding data fields that increase the data integrity burden.  Was this needed for something?  An auxiliary data structure (outside of `Tree and TreeNode`) linking nodes to trees would achieve what you need.