如何在 BTreeMap/BTreeSet 中找到下一个较小的键？

Question

如果我正确理解 b 树，那么在对数时间内搜索密钥应该很容易且可能。如果key不存在，可以返回下一个越来越大的key；给定键的邻居（如果它会被插入）。

此功能是否已经存在？

使用当前 API 的一种可能但复杂的方法是插入密钥，然后获取该密钥的迭代器，以便我们可以在此迭代器上调用 next。虽然，它也不清楚如何获得一个新插入元素的迭代器（参见this question）

为什么缺少这些方法或者我缺少什么？

Answer 1

您可以在返回的Range 对象上使用range method 和迭代器方法：

use std::collections::BTreeMap;
let mut map = BTreeMap::new();
map.insert(2, 0);
map.insert(3, 1);
map.insert(5, 2);
map.insert(7, 3);
map.insert(11, 4);
let key = 6;
println!("maximum in map less than {}: {:?}",
         key, map.range(..key).next_back().unwrap());
println!("minimum in map greater than {}: {:?}",
         key, map.range(key..).next().unwrap());

next_back() 和 next() 都执行树遍历，因此它们的效率相当高。