Java

java treenode如何实现节点排序

小樊
81
2024-11-28 08:28:39
栏目: 编程语言

在Java中,你可以使用TreeMapTreeSet来实现对树节点的排序。这两种数据结构都是基于红黑树实现的,它们可以保证元素的自然顺序或者根据自定义的比较器进行排序。

以下是一个使用TreeMap实现节点排序的示例:

import java.util.Comparator;
import java.util.Map;
import java.util.TreeMap;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }
}

public class SortedTreeNode {
    public static void main(String[] args) {
        TreeNode root = new TreeNode(10);
        root.left = new TreeNode(5);
        root.right = new TreeNode(15);
        root.left.left = new TreeNode(3);
        root.left.right = new TreeNode(7);
        root.right.left = new TreeNode(12);
        root.right.right = new TreeNode(18);

        // 使用TreeMap对树节点进行排序
        Map<Integer, TreeNode> sortedNodes = new TreeMap<>(Comparator.comparingInt(TreeNode::getVal));
        sortedNodes.putAll(buildNodeMap(root));

        // 输出排序后的节点
        for (Map.Entry<Integer, TreeNode> entry : sortedNodes.entrySet()) {
            System.out.println("Value: " + entry.getKey() + ", Node: " + entry.getValue());
        }
    }

    private static Map<Integer, TreeNode> buildNodeMap(TreeNode node) {
        if (node == null) {
            return new TreeMap<>();
        }

        Map<Integer, TreeNode> nodeMap = buildNodeMap(node.left);
        nodeMap.put(node.val, node);
        nodeMap.putAll(buildNodeMap(node.right));

        return nodeMap;
    }
}

在这个示例中,我们首先创建了一个TreeNode类,然后使用TreeMap对树节点进行排序。buildNodeMap方法是一个递归方法,它将树节点转换为一个以节点值为键,节点为值的TreeMap。最后,我们遍历排序后的TreeMap并输出节点的值和节点对象。

如果你想要根据自定义的比较器对树节点进行排序,可以在创建TreeMap时传入一个Comparator实例。例如,以下示例将根据节点的值降序排序:

Map<Integer, TreeNode> sortedNodes = new TreeMap<>(Comparator.reverseOrder());

0
看了该问题的人还看了