从 TreeMap 返回相邻键

Question

鉴于以下TreeMap：

Map<Double, Integer> treeMap = new TreeMap<Double, Integer>() {{. 
        put("52.1", 1);
        put("53.4", 2); 
        put("57.1", 3); 
        put("59.4", 7); 
        put("60.2", 11); 
        put("71.6", 16)}};

对于给定的 double 返回最近的 n 个匹配项（在两个方向上）的最佳方法是什么？例如，n=2 和 "58.0" 将返回 53.4、57.1、59.4、60.2

Answer 1

tl;博士

new TreeMap <>(
        Map.of(
                52.1d , 1 ,
                53.4d , 2 ,
                57.1d , 3 ,
                59.4d , 7 ,
                60.2d , 11 ,
                71.6d , 16
        )
)
.floorKey( 58.0d )     // Or call `.ceilingKey( 58.0d )`

看到这个code run live at IdeOne.com。

57.1

NavigableMap::lowerKey/higherKey

TreeMap 是 NavigableMap。

NavigableMap 接口提供检索相邻键（和条目）的方法。

• lowerKey
• higherKey

重复命令以获取尽可能多的相邻键，直到收到null。

比较键

二读时，您似乎没有钥匙。您有一个值要与地图中的键进行比较。

在这种情况下，获取映射中所有键的Set。 NavigableMap 扩展 SortedMap。所以我们可以拨打SortedMap::keySet。返回的集合按排序顺序迭代。

Set< Double > set = map.keySet() ;

为此创建一个List，以按索引访问各个元素。

List< Double > doublesListSorted = List.of( set.toArray() ) ;  // Get un-modifiable list.

现在您可以循环排序列表以比较值。

NavigableMap::floorKey/ceilingKey

或者，作为dnault commented，我们可以要求NavigableMap 比较键值以进行最近匹配。

• floorKey
  返回小于或等于给定键的最大键，如果没有这样的键，则返回 null。
• ceilingKey
  返回大于或等于给定键的最小键，如果没有这样的键，则返回 null。

示例代码

制作输入地图。我们最初使用Map.of 是因为它方便的文字语法。然后我们将un-modifiable map 提供给TreeMap 的构造函数。我们在这里看到的文字输入恰好是排序的，但这无关紧要，因为TreeMap 构造函数将按键排序。

NavigableMap < Double, Integer > map =
        new TreeMap <>(
                Map.of(
                        52.1d , 1 ,
                        53.4d , 2 ,
                        57.1d , 3 ,
                        59.4d , 7 ,
                        60.2d , 11 ,
                        71.6d , 16
                )
        );

map.toString(): {52.1=1, 53.4=2, 57.1=3, 59.4=7, 60.2=11, 71.6=16}

设置我们想要最接近匹配的目标。

Double target = 58.0d;

获取相邻的键，下一个更低，下一个更高。

Double nextLower = map.floorKey( target );
Double nextHigher = map.ceilingKey( target );

nextLower = 57.1 下一个更高 = 59.4

看到这个code run live at IdeOne.com.

---

注意事项：

• 作为commented by erickson，你的双括号初始化有点像黑客。鉴于此答案中显示的新 List.of 文字语法，现在没有必要了。
• 如果您关心accuracy，请使用BigDecimal 而不是Double。

---