如何按包含的枚举对对象列表进行排序？答案

【问题标题】：How to sort a list of objects by contained enum?如何按包含的枚举对对象列表进行排序？
【发布时间】：2021-06-07 09:54:37
【问题描述】：

我正在使用 java 8。我有一个汽车对象列表。我想按特定顺序对它们进行排序。
每个汽车对象都属于一个模型。我喜欢按型号对汽车对象列表进行排序。排序后的列表应按以下顺序排列，车型为 SEDAN 的汽车，然后是 BMW，然后是 UNICORN。

import java.util.ArrayList;
import java.util.List;

enum Model {

    BMW, SEDAN, UNICORN

}

public class Car {

    private Model model;

    private int id;

    public Car(Model model, int id) {
        super();
        this.model = model;
        this.id = id;
    }

    public Model getModel() {
        return model;
    }

    public void setModel(Model model) {
        this.model = model;
    }

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public static void main(String[] args) {

        List<Car> l = new ArrayList<Car>();

        Car c1 = new Car(Model.BMW, 1);
        Car c2 = new Car(Model.BMW, 2);
        Car c3 = new Car(Model.SEDAN, 3);
        Car c4 = new Car(Model.SEDAN, 4);
        Car c5 = new Car(Model.SEDAN, 5);
        Car c6 = new Car(Model.UNICORN, 6);
        Car c7 = new Car(Model.BMW, 7);
        Car c8 = new Car(Model.BMW, 8);

        l.add(c1);
        l.add(c2);
        l.add(c3);
        l.add(c4);
        l.add(c5);
        l.add(c6);
        l.add(c7);
        l.add(c8);
    }
}

所以当我打印排序列表时，它应该是这样的

[Car [model=SEDAN, id=3],  Car [model=SEDAN, id=4]  ,Car [model=SEDAN, id=5],  [model=BMW, id=1], Car [model=BMW, id=2], Car [model=BMW, id=7], Car [model=BMW, id=8],  Car [model=UNICORN, id=6]]

如何使用自定义比较器在模型枚举上实现这种排序顺序。

如果你能帮忙，感激不尽

【问题讨论】：

标签： java collections java-8 enums comparator

【解决方案1】：

您可以将 int 值分配给枚举，并根据您想要的偏好给它们赋值。

然后为您的Car 对象实现Comparable 接口，以使用这些分配给每个模型的int 值进行排序

代码

import java.util.ArrayList;
import java.util.List;
// assign enum values in ascending order here 
enum Model {
    SEDAN(0),
    BMW(1),
    UNICORN(2);

    private int value;

    private Model(int value) {
        this.value = value;
    }

    public int getValue() {
        return value;
    }

    public void setValue(int value) {
        this.value = value;
    }
}
// implement comparable class 
public class Car implements Comparable<Car> {

    private Model model;

    private int id;

    public Car(Model model, int id) {
        super();
        this.model = model;
        this.id = id;
    }

    public Model getModel() {
        return model;
    }

    public void setModel(Model model) {
        this.model = model;
    }

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }
   // compare Cars by Model int value
    @Override
    public int compareTo(Car otherCar) {
        if (this.getModel().getValue() > otherCar.getModel().getValue())
            return 1;
        else return -1;
    }

    @Override
    public String toString() {
        return "Car{" +
                "model=" + model +
                ", id=" + id +
                '}';
    }
}

public static List<Car> l = new ArrayList<Car>();
Car c1 = new Car(Model.BMW, 1);
Car c2 = new Car(Model.BMW, 2);
Car c3 = new Car(Model.SEDAN, 3);
Car c4 = new Car(Model.SEDAN, 4);
Car c5 = new Car(Model.SEDAN, 5);
Car c6 = new Car(Model.UNICORN, 6);
Car c7 = new Car(Model.BMW, 7);
Car c8 = new Car(Model.BMW, 8);

l.add(c1);
l.add(c2);
l.add(c3);
l.add(c4);
l.add(c5);
l.add(c6);
l.add(c7);
l.add(c8);
l.sort(Car::compareTo);
l;

输出

[Car{model=SEDAN, id=5}, 
Car{model=SEDAN, id=4}, 
Car{model=SEDAN, id=3}, 
Car{model=BMW, id=8},
Car{model=BMW, id=7}, 
Car{model=BMW, id=2}, 
Car{model=BMW, id=1}, 
Car{model=UNICORN, id=6}]

【讨论】：

【解决方案2】：

第一种方法

重新排序枚举条目，使结果顺序符合您的要求：

enum Model {
    SEDAN, BMW, UNICORN
}

然后您可以按如下方式对列表进行排序：

l.sort(Comparator.comparing(Car::getModel));

请注意，枚举值是按它们在代码中的顺序进行比较的（即ordinal()）

第二种方法

为每个枚举条目赋值：

enum Model {
    SEDAN(0), BMW(1), UNICORN(2);
    final int val;
    Model(int val) { this.val = val; }
}

然后您可以按如下方式对列表进行排序：

l.sort(Comparator.comparing((Car car) -> car.getModel().val));

【讨论】：