TreeSet未使用自定义比较器正确排序

时间:2015-02-09 17:07:25

标签: java

我有一个自定义的TreeSet:

    TreeSet<String> sortedOptionSet = new TreeSet<String> (new Comparator<String>() {

        @Override
        public int compare(String arg1, String arg2) {
            if (arg1 == null || arg2 == null) {
                return -1;
            }

            String o1 = arg1, o2 = arg2;
            String k1 = "1", k2 = "2";

            try{
                if (o1.contains("P1")) {
                    return -1;
                }
                else if (o2.contains("P1")) {
                    return 1;
                }
                else if (o1.contains("P2")) {
                    return -1;
                }
                else if (o2.contains("P2")) {
                    return 1;
                }
            } catch (Exception e) {}


            //If there is no rule for this combination, order them using key number ascending
            int result = 1;
            try {
                Integer key1 = Integer.valueOf(k1);
                Integer key2 = Integer.valueOf(k2);
                result = key1.compareTo(key2);
                if (result == 0) {
                    result = 1;
                }
            } catch (Exception e) {
            }
            return result;

        }
    });

    sortedOptionSet.add("TEST1");
    sortedOptionSet.add("P1");
    sortedOptionSet.add("TEST2");
    sortedOptionSet.add("P2");
    sortedOptionSet.add("TEST3");

    ArrayList<String> result = new ArrayList<String>();
    result.addAll(sortedOptionSet);

    for (String s : result) {
        System.out.println(s);
    }

我把P1的条件放在第一位,并认为结果应该是:

P1
P2
TEST1
TEST2
TEST3

但结果返回

P2
P1
TEST1
TEST2
TEST3

我无法弄清楚为什么这是行为。请帮忙。

1 个答案:

答案 0 :(得分:6)

您的比较器违反了许多合同规则。特别是,比较器应该是一致的:

A > B iff B < A
A = B iff B = A
A > B and B > C ==> A > C

事实并非如此。例如,如果A和B都为空,compare(A, B)将导致A&lt; B和compare(B, A)将导致B&lt;甲

如果A和B都包含P1或P2,则相同。

如果它们具有相同的整数键,compare(A, B)将导致A&gt; B,compare(B, A)将导致B&gt;答:除非在这种情况下,你甚至不比较比较器的参数,而是两个硬编码值k1和k2。如果有一个异常解析任何一个数字,第一个总是最大的。

相关问题