We need to add a customized constructor to accept TreeNode of root in the given code sample.

public Solution(TreeNode root) {
    initialize(root);
  }