如何使用比较器定义自定义排序顺序？

sorting enums java

2022-08-31 09:50:19

我想为汽车列表开发一个排序演示。我正在使用数据表来显示汽车列表。现在实际上我想按汽车颜色对列表进行排序。这里它不是按字母顺序排序的。我想使用我的自定义排序顺序，例如红色汽车先来，然后蓝色，等等。

为此，我尝试使用Java Comparator和Compeable，但它只允许按字母顺序排序。

因此，任何人都可以指导我实现要使用的技术的方法，以便排序变得更快。

class Car implements Comparable<Car>
{
    private String name;
    private String color;

    public Car(String name, String color){
        this.name = name;
        this.color = color;
    }

    //Implement the natural order for this class
    public int compareTo(Car c) {
        return name.compareTo(c.name);
    }

    static class ColorComparator implements Comparator<Car> {
        public int compare(Car c1, Car c2) {
            String a1 = c1.color;
            String a2 = c2.color;
            return a1.compareTo(a2);
        }
    }

    public static void main(String[] args) {
        List<Car> carList = new ArrayList<>();
        List<String> sortOrder = new ArrayList<>();

        carList.add(new Car("Ford","Silver"));
        carList.add(new Car("Tes","Blue"));
        carList.add(new Car("Honda","Magenta"));

        sortOrder.add("Silver");
        sortOrder.add("Magenta");
        sortOrder.add("Blue");

        // Now here I am confuse how to implement my custom sort             
    }
}

答案 1

我建议你为你的汽车颜色创建一个枚举，而不是使用字符串，枚举的自然顺序将是你声明常量的顺序。

public enum PaintColors {
    SILVER, BLUE, MAGENTA, RED
}

和

 static class ColorComparator implements Comparator<CarSort>
 {
     public int compare(CarSort c1, CarSort c2)
     {
         return c1.getColor().compareTo(c2.getColor());
     }
 }

您将字符串更改为“画色”，然后在主列表中，您的汽车列表变为：

carList.add(new CarSort("Ford Figo",PaintColor.SILVER));

...

Collections.sort(carList, new ColorComparator());

答案 2

怎么样：

List<String> definedOrder = // define your custom order
    Arrays.asList("Red", "Green", "Magenta", "Silver");

Comparator<Car> comparator = new Comparator<Car>(){

    @Override
    public int compare(final Car o1, final Car o2){
        // let your comparator look up your car's color in the custom order
        return Integer.valueOf(
            definedOrder.indexOf(o1.getColor()))
            .compareTo(
                Integer.valueOf(
                    definedOrder.indexOf(o2.getColor())));
    }
};

原则上，我同意使用是一种更好的方法，但这个版本更灵活，因为它允许您定义不同的排序顺序。enum

更新

番石榴在其订购类中融入了此功能：

List<String> colorOrder = ImmutableList.of("red","green","blue","yellow");
final Ordering<String> colorOrdering = Ordering.explicit(colorOrder);
Comparator<Car> comp = new Comparator<Car>() {
    @Override
    public int compare(Car o1, Car o2) {
        return colorOrdering.compare(o1.getColor(),o2.getColor());
    }
};

这个版本有点不那么冗长。

再次更新

Java 8 使 Comparator 更加不冗长：

Comparator<Car> carComparator = Comparator.comparing(
        c -> definedOrder.indexOf(c.getColor()));