1469. Find All The Lonely Nodes
in Coding Interview on Easy, Tree, Recursion
이진트리에서 부모 노드의 유일한 자식 노드 찾기
1469. Find All The Lonely Nodes
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
ArrayList<Integer> nodes = new ArrayList<>();
// 부모 노드의 유일한 자식 노드 찾기
// T: O(n)
public List<Integer> getLonelyNodes(TreeNode root) {
// 루트 노드는 제외한다.
addLonelyNodes(root, false);
return nodes;
}
// 왼쪽/오른쪽 노드를 재귀로 순회하면서 자식노드가 하나인 노드를 결과 리스트에 추가
private void addLonelyNodes(TreeNode node, boolean lonely) {
// 종료조건
if (node == null) return;
// 결과저장
if (lonely) nodes.add(node.val);
// 왼쪽노드 처리
addLonelyNodes(node.left, node.right == null);
// 오른쪽노드 처리
addLonelyNodes(node.right, node.left == null);
}
}