FilesExpand file tree

 master

/

TreeToForestByErasingNodes.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

29 lines (23 loc) · 709 Bytes

 master

/

TreeToForestByErasingNodes.java

Copy path

Top

File metadata and controls

• Code
• Blame

29 lines (23 loc) · 709 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

/**

*

*/

public class TreeToForestByErasingNodes {

// shouldBeErased is predefined

public List<Node> treeToForest(Node root) {

List<Node> res = new ArrayList<>();

dfs(root, true, res);

return res;

}

private Node dfs(Node curr, boolean parentIsErasable, List<Node> res) {

if (curr == null) return null;

boolean currIsErasable = shouldBeErased(curr);

if (parentIsErasable && !currIsErasable) res.add(curr);

curr.left = dfs(curr.left, currIsErasable, res);

curr.right = dfs(curr.right, currIsErasable, res);

return currIsErasable ? null : curr;

}

class Node {

Node left;

Node right;

}

}