Sortowanie własnych typów. :: 4programmers.net

Rejestracja:około 13 lat
Ostatnio:ponad 2 lata
Lokalizacja:Wrocław
Postów:259

Cześć.
Mam własny typ który składa się z dwóch intów. Chciałbym go posortować najpierw wg pierwszego inta, a następnie sortować "wewnętrznie" wg drugiego inta. Wyglądało by to tak. Weźmy pray: (2,4),(1,3),(1,2),(2,2),(3,1),(3,0) Najpierw sortujemy wg pierwszego i dostajemy coś takiego: (1,3),(1,2),(2,4),(2,2),(3,1),(3,0) Następnie wg drugiego ```
(1,2),(1,3),(2,2),(2,4),(3,0),(3,1)

Kopiuj

```java
public class Test {

    private int a;
    private int b;

    public Test(int a, int b) {
        this.a = a;
        this.b = b;
    }

    public int getA() {
        return a;
    }

    public int getB() {
        return b;
    }

    @Override
    public String toString() {
        return a + " " + b;
    }

    public static void main(String[] args) {
        List<Test> testList = new LinkedList<>();

        ThreadLocalRandom current = ThreadLocalRandom.current();

        for (int i = 0; i < 10; i++) {
            testList.add(new Test(current.nextInt(0, 10), current.nextInt(0, 10)));
        }

        testList.forEach(System.out::println);

        testList.sort(Comparator.comparingInt(Test::getA));

        System.out.println("---");
        testList.forEach(System.out::println);
    }

Patryk27

2017-04-20 17:58

Patryk272017-04-20 17:58

Moderator

Rejestracja:ponad 17 lat
Ostatnio:ponad rok
Lokalizacja:Wrocław
Postów:13042

Napisz własną metodę porównującą (własny comparator).

edytowany 1x, ostatnio: Patryk27 2017-04-20 17:58

barslo

2017-04-20 18:06

barslo2017-04-20 18:06

Rejestracja:około 13 lat
Ostatnio:ponad 2 lata
Lokalizacja:Wrocław
Postów:259

Eh tyle męczenia, aż tu nagle eureka :)

samouczek

2017-04-20 18:12

samouczek2017-04-20 18:12

Rejestracja:około 8 lat
Ostatnio:ponad 7 lat
Postów:67

Niech Twoja klasa zaimplementuje interfejs Comparable. Przykład poniżej:

Kopiuj

@Override
public int compareTo(Test other) {
    if (other == null) {
        return 1;
    }
    if (a == other.a) {
        return b - other.b;
    }
    return a - other.a;
}

Wtedy będziesz mógł sortować te obiekty używając Collections.sort bez podawania "komparatora". Jeśli wolisz wersję z komparatorem, implementacja może być podobna do metody wyżej.

edytowany 1x, ostatnio: samouczek 2017-04-20 18:14

barslo

2017-04-20 18:13

barslo2017-04-20 18:13

Rejestracja:około 13 lat
Ostatnio:ponad 2 lata
Lokalizacja:Wrocław
Postów:259

Zrobiłem tak

Kopiuj

testList.sort((o1, o2) -> {
            if (o1.getA() == o2.getA())
                return o1.getB() - o2.getB();
            else
                return o1.getA() - o2.getA();
        });