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

Question

我想用字符串对映射进行排序，以便某些字符串具有优先级，其余字符串则照常排序。

像这样：

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

就像在 this 问题中一样。

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

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);
}

如您所见，对于优先级更高的字符串，这会变得相当麻烦。

有更好的方法吗？

解决方案（感谢amit提供的思路）：使用第二张地图存储优先级：

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 时，才返回到常规字符串的 compareTo()。

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

(1) 确保两个元素首先存在于 prio 中，如果一个不存在 - 解决。

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

Best way to prioritize certain Strings while sorting a Map

java

string

sorting

compare

treemap