Пользовательский компаратор TreeSet Algo .. Сравнение строк

Question

Из предоставленной входной строки:

{"200,400,7,1", "100,0,1,1", "200,200,3,1", "0,400,11,1", "407,308,5,1", "100,600,9,1" },

Я добавляю то же самое в TreeSet и хочу, чтобы он был отсортирован с 3-м порядком элементов, поэтому ожидаемый результат будет:

(100,0,1,1) (200,200,3,1) (407,308,5,1) (200,400,7,1) (100,600,9,1) (0,400,11,1)

Но мой фактический результат:

(100,0,1,1) (0,400,11,1) (200,200,3,1) (407,308,5,1) (200,400,7,1) (100,600,9,1)

Но так как сравнение строк 11 меньше 9, но с точки зрения целого числа, 11>9 . Мой ожидаемый результат становится другим. Предложите мне идею, чтобы решить то же самое.

import java.util.Comparator;
import java.util.TreeSet;

public class TreeSetComparator {

    public static void main(String[] args) {
        Comparator<String> comparator = new Comparator<String>() {
            @Override
            public int compare(String a, String b) {
                String aStr = a;
                String bStr = b;
                String[] splitA = aStr.split(",");
                String[] splitB = bStr.split(",");

                return splitA[2].compareTo(splitB[2]);
            }
        };

        String[] arr = { "200,400,7,1", "100,0,1,1", "200,200,3,1",
                "0,400,11,1", "407,308,5,1", "100,600,9,1" };

        TreeSet<String> ts = new TreeSet<String>(comparator);
        for (String str : arr) {
            ts.add(str);
        }

        for (String element : ts)
            System.out.print(element + " ");

    }
}

Answer 1

Вы сортируете в лексикографическом порядке ("123" приходит раньше "20"), что вам нужно сделать, это преобразовать их в целые числа, а затем сравнить их:

Не:

return splitA[2].compareTo(splitB[2]);

но:

return Integer.valueOf(splitA[2]).compareTo(Integer.valueOf(splitB[2]));

Тем не менее, несколько более чистым способом было бы создать пользовательский объект, содержащий эти 4 различных значения, а затем создать Comparator который сравнивает 3-^е значение такого объекта:

Следующие:

public class Main {
    public static void main (String[] args) {

        Comparator<CustomObject> sortOn3rdValue = new Comparator<CustomObject>() {
            @Override
            public int compare(CustomObject o1, CustomObject o2) {
                return o1.v3 < o2.v3 ? -1 : o1.v3 > o2.v3 ? 1 : 0;
            }
        };

        Set<CustomObject> objects = new TreeSet<CustomObject>(sortOn3rdValue);

        String[] arr = { "200,400,7,1", "100,0,1,1", "200,200,3,1", "0,400,11,1", "407,308,5,1", "100,600,9,1" };

        for(String a : arr) {
            objects.add(new CustomObject(a.split(",")));
        }

        for(CustomObject co : objects) {
            System.out.println(co);
        }
    }
}

class CustomObject {

    final int v1, v2, v3, v4;

    CustomObject(String[] strValues) {
        // assume strValues.lenght == 4
        v1 = Integer.valueOf(strValues[0]);
        v2 = Integer.valueOf(strValues[1]);
        v3 = Integer.valueOf(strValues[2]);
        v4 = Integer.valueOf(strValues[3]);
    }

    @Override
    public String toString() {
        return String.format("(%d,%d,%d,%d)", v1, v2, v3, v4);
    }
}

напечатал бы:

(100,0,1,1)
(200,200,3,1)
(407,308,5,1)
(200,400,7,1)
(100,600,9,1)
(0,400,11,1)