在对 Map 进行排序时确定某些字符串优先级的最佳方法

Question

我想用字符串对地图进行排序，以便对某些字符串进行优先级排序，其余字符串像往常一样排序。

喜欢这个：

"Dan",   "value"  //priority 1
"Eric",  "value"  //priority 2
"Ann",   "value"  //priority 3
"Bella", "value"  //no priority
"Chris", "value"  //no priority

就像这个问题一样。

我

正在使用树状图，我当前的比较方法如下所示：

public int compare(String o1, String o2) {
    if (o1.equals(o2)) return 0;
    if (o1.equals("Dan")) return -1;
    if (o2.equals("Dan")) return 1;
    if (o1.equals("Eric")) return -1;
    if (o2.equals("Eric")) return 1;
    if (o1.equals("Ann")) return -1;
    if (o2.equals("Ann")) return 1;
    else return o1.compareTo(o2);
}

如您所见，使用优先级更高的字符串会变得非常麻烦。

有没有更好的方法可以做到这一点？

---

解决方案（感谢阿米特的想法）：使用第二个地图来存储优先级：

TreeMap<String, Integer> prio = new TreeMap<>();
prio.put("Dan", 1);
prio.put("Eric", 2);
prio.put("Ann", 3);
comparator = new Comparator<String>() {
    @Override
    public int compare(String o1, String o2) {
        if (prio.containsKey(o1)) {
            if (prio.containsKey(o2)) {
                return prio.get(o1).compareTo(prio.get(o2));
            } else return -1;
        } else if (prio.containsKey(o2)) {
            return 1;
        } else return o1.compareTo(o2);
    }
};

Answer 1

使用第二张地图：

Map<String,Integer> prio，其中值是每个字符串的优先级。

在您的比较器中 - 首先根据prio.get(o1).compareTo(prio.get(o2)) ¹ 进行比较，只有当结果为 0 时，才回退到常规 String 的compareTo()。

创建

map后prio不会更改非常重要，否则您的地图将完全混乱，无法正确查找和插入元素。

---

（1）确保两个元素首先存在于prio，如果不存在 - 解决。