Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- public void flatten(TreeNode root) {
- root = flattenRecursive(root)[0];
- }
- private TreeNode[] flattenRecursive(TreeNode node) {
- if (node.left == null && node.right == null ) return new TreeNode[] {node, node};
- TreeNode[] nodesLeft = flattenRecursive(node.left);
- TreeNode[] nodesRight = flattenRecursive(node.right);
- nodesLeft[1].right = nodesRight[0];
- node.right = nodesLeft[0];
- node.left = null;
- return new TreeNode[] { node, nodesRight[1]};
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement