Java 中二叉树奇数位置和偶数位置节点之和的差异
javaobject oriented programmingprogramming更新于 2024/4/20 23:46:00
问题陈述
对于给定的二叉树，编写一个程序来找出奇数位置和偶数位置节点之和的差异。假设根节点在第 0 级，奇数位置，根节点的左/右子节点在第 2 级，左子节点在奇数位置，右子节点在偶数位置，依此类推。
示例
5 / \ 2 6 / \ \ 1 4 8 / / \ 3 7 9 奇数位置节点之和 = 5 + 2 + (1 + 8) + (3 + 9) = 5 + 2 + 9 + 12 = 28 偶数级别节点之和 = 0 + 6 + 4 + 7 = 17 差 = 11。
解决方案
使用级别顺序遍历。遍历期间，将第一个元素标记为奇数位置，然后在遇到新元素时切换到偶数位置，然后切换回下一个，依此类推。
示例
以下是 Java 中查找所需输出的程序。
import java.util.LinkedList;
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(LinkedList<Node> queue){
if(queue.isEmpty()) return 0;
int evenSum = 0;
int oddSum = 0;
while(true){
int nodes = queue.size();
if(nodes == 0) break;
boolean isOdd = true;
while(nodes > 0){
Node node = queue.peek();
if(isOdd) oddSum += node.data;
else evenSum += node.data;
queue.remove();
nodes--;
if(node.left != null) queue.add(node.left);
if(node.right != null) queue.add(node.right);
isOdd = !isOdd;
}
}
return oddSum - evenSum;
}
public static void main(String args[]){
Node tree = getTree();
LinkedList<Node> queue = new LinkedList<Node>();
queue.add(tree);
System.out.println(difference(queue));
}
}
输出
11
