Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How can I get the last item in a BTreeMap?

Tags:

rust

b-tree

If you have a sorted map of key/value pairs (or just keys), one of the obvious operations is to get the first or last pair (or key).

C++'s std::vector has front() and back() for this purpose. std::map doesn't, but *map.begin() and *map.rbegin() (reverse iterator) work for this (assuming one knows the map is not empty).

In Rust, getting the first element of a map seems to require map.iter().next().unwrap() — ugly, but perhaps justified considering some error checking is needed.

How can we get the last element? By stepping over all elements: map.iter().last().unwrap()?

I see that there is Iterator::rev(), so is map.iter().rev().next().unwrap() a reasonable alternative?

like image 667
dhardy Avatar asked Nov 13 '15 18:11

dhardy


2 Answers

btree_map::Iter, which is returned by BTreeMap::iter(), implements DoubleEndedIterator, so indeed, either the approach with rev() would work or you can use the next_back() method directly:

let (key, value) = map.iter().next_back().unwrap();
like image 80
Vladimir Matveev Avatar answered Oct 24 '22 19:10

Vladimir Matveev


https://github.com/rust-lang/rust/issues/31690#issuecomment-184445033

A dedicated method would improve discoverability, but you can do:

let map: BTreeMap<K, V> = ...;
let min = map.iter().next();
let max = map.iter().next_back();
and the same for BTreeSet.
like image 41
rubyu2 Avatar answered Oct 24 '22 20:10

rubyu2