문제 설명
주어진 이진 트리를 사용하여 홀수 레벨과 짝수 레벨의 노드 합 사이의 차이를 찾는 프로그램을 작성하십시오. 레벨 1에서 루트, 레벨 2에서 루트의 왼쪽/오른쪽 자식 등을 가정합니다.
예시
5 / \ 2 6 / \ \ 1 4 8 / / \ 3 7 9 Sum of nodes at odd level = 5 + 1 + 4 + 8 = 18 Sum of nodes at even level = 2 + 6 + 3 + 7 + 9 = 27 Difference = -9.
해결책
재귀 순회를 사용합니다. 순회하는 동안 루트 노드와 왼쪽 및 오른쪽 자식의 차이를 반환합니다.
예시
다음은 필요한 출력을 찾는 Java 프로그램입니다.
class Node {
int data;
Node left, right;
Node(int data){
this.data = data;
this.left = this.right = null;
}
}
public class JavaTester {
public static Node getTree(){
Node root = new Node(5);
root.left = new Node(2);
root.right = new Node(6);
root.left.left = new Node(1);
root.left.right = new Node(4);
root.left.right.left = new Node(3);
root.right.right = new Node(8);
root.right.right.right = new Node(9);
root.right.right.left = new Node(7);
return root;
}
public static int difference(Node node){
if(node == null) return 0;
return node.data - difference(node.left) - difference(node.right);
}
public static void main(String args[]){
Node tree = getTree();
System.out.println(difference(tree));
}
} 출력
-9