How can I find the diameter of a binary tree with a restriction that the end nodes should be on a different level (have different height)?

Question

Here\'s a link to the question on leetcode of finding a diameter of a binary tree that I know how to solve. But how can I do this tricky modification, shall I pass a struct