Java

java treenode如何实现排序

小樊
81
2024-11-28 10:21:47
栏目: 编程语言

在Java中,你可以通过实现Comparable接口或提供一个Comparator来实现对树节点的排序。下面是两种方法的示例:

方法1:实现Comparable接口

class TreeNode implements Comparable<TreeNode> {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }

    @Override
    public int compareTo(TreeNode other) {
        return Integer.compare(this.val, other.val);
    }
}

在这个例子中,TreeNode类实现了Comparable接口,并重写了compareTo方法。这样,当你将TreeNode对象添加到TreeSetTreeMap等集合时,它们会自动根据值进行排序。

方法2:提供一个Comparator

import java.util.Comparator;

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

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

class TreeNodeComparator implements Comparator<TreeNode> {
    @Override
    public int compare(TreeNode node1, TreeNode node2) {
        return Integer.compare(node1.val, node2.val);
    }
}

在这个例子中,我们创建了一个名为TreeNodeComparator的类,它实现了Comparator接口,并重写了compare方法。然后,你可以使用这个比较器对TreeNode对象进行排序,例如:

List<TreeNode> nodes = new ArrayList<>();
nodes.add(new TreeNode(3));
nodes.add(new TreeNode(1));
nodes.add(new TreeNode(2));

Collections.sort(nodes, new TreeNodeComparator());

这将根据节点的值对nodes列表进行排序。

0
看了该问题的人还看了