I would like to store a mapping of an integer
key to a float
value in-memory.
I have roughly 130 million keys (and, accordingly, 130 million values).
My focus is on lookup performance -- I have to do many, many millions of lookups.
The C++ STL library has a map
class for associative arrays of this sort. I have several questions about map
.
What is the storage overhead of map
for a dataset of the size mentioned above? How does storage overhead scale, in general, with map
?
It looks like the underlying data structure for map
is a red-black, balanced binary tree. It sounds like the real-world performance for this is O(log n)
for insertion and retrieval.
It mentions O(1)
for a hinted insertion. My input is pre-sorted, so I believe I should be able to provide a hint for insertion events. How would I provide this hint, using the methods listed here?
Is there an STL container that provides better lookup performance?
Are there other publicly-available, open-source frameworks with an associate array class that uses an underlying data structure that would perform better than STL map
?
If writing my own container class would provide better lookup performance, what data structures might I research?
I am using GCC 4 for this task, running under either Linux or Mac OS X.
I apologize in advance if these are dumb questions. Thank you for your advice.
As a partial answer to your question concerning lookup performance, you have to consider your insertion pattern. You noted that
std::map
uses a Red-Black Tree as a hedge against linearizing a carefully sorted insertion into a linked list. Hence, such a tree provides O(log n) lookup time despite an aberrant insertion order. You pay a cost for this, however, in insertion, removal, and traversal performance, as well as losing locality of reference for repeated reading of "nearby" data.A hash table may offer faster lookup if you can tune a hash function for your key type (an integer, you say) that will preclude collisions. If your data set was fixed, such that you could load it once and only read it afterward, you could use parallel arrays of integers and floats, and use
std::lower_bound
to find your match via binary search. Sorting the parallel arrays properly would be a chore, if your keys are divorced from their corresponding values, but you'd enjoy tighter storage and locality of reference than storing an array of, say,std::pair
.Considering the vast amount of memory used, you must also consider that any memory access in searching will result in a memory cache fault.
In this regard a mixed solution of a small hashmap as first layer and sorted vectors for the buckets is probably the best.
The idea is to keep the hashtable index in the cache memory, and to search in smaller sorted containers to reduce the number of cache faults.
A vector is absolutely going to kill a map here assuming you don't need to do insertions in the middle of the vector. I wrote a custom allocator to track memory usage, and here are the results in Visual Studio 2005:
std::map<int, float>:
std::vector<std::pair<int, float> >:
std::map is using over twice the storage and taking 20 times longer to insert all the items.
Most compilers ship with a non-standard (but working)
hash_map
(orunordered_map
) that might be faster for you. It's coming in C++0x (is in tr1) and it is also (as always) in boost already.GCC did too, but I haven't done C++ on that for .. 12 years .., but it should still be in there somewhere.
Given what you've said, I'd think very hard about using an
std::vector<pair<int, float> >
, and usingstd::lower_bound
,std::upper_bound
, and/orstd::equal_range
to look up values.While the exact overhead of
std::map
can (and does) vary, there's little or no room for question that it will normally consume extra memory and look up values more slowly than a binary search in a vector. As you've noted, it's normally (and almost unavoidably) implemented as some sort of balanced tree, which imposes overhead for the pointers and the balancing information, and typically means each node is allocated separately as well. Since your nodes are pretty small (typically 8 bytes) that extra data is likely to be at least as much as what you're actually storing (i.e. at least 100% overhead). Separate allocations often mean poor locality of reference, which leads to poor cache usage.Most implementations of
std::map
use a red-black tree. If you were going to use anstd::map
, an implementation that uses an AVL tree would probably suit your purposes better -- an AVL tree has slightly tighter constraints on balancing. This gives slightly faster lookup at the expense of slightly slower insertion and deletion (since it has to re-balance more often to maintain its stricter interpretation of "balanced"). As long as your data remains constant during use, however, anstd::vector
is still almost certainly better.One other possibility worth noting: if your keys are at least fairly even distributed, you might want to try looking up using interpolation instead of bisection. i.e. instead of always starting at the middle of the vector, you do a linear interpolation to guess at the most likely starting point for the lookup. Of course, if your keys follow some known non-linear distribution, you can use a matching interpolation instead.
Assuming the keys are reasonably evenly distributed (or at least follow some predictable pattern that's amenable to interpolation), the interpolation search has a complexity of O(log log N). For 130 million keys, that works out to around 4 probes to find an item. To do significantly better than that with (normal/non-perfect) hashing, you need a good algorithm, and you need to keep the load factor in the table quite low (typically around 75% or so -- i.e. you need to allow for something like 32 million extra (empty) spots in your table to improve the expected complexity from four probes to three). I may just be old fashioned, but that strikes me as a lot of extra storage to use for such a small speed improvement.
OTOH, it's true that this is nearly the ideal situation for perfect hashing -- the set is known ahead of time, and the key is quite small (important, since hashing is normally linear on the key size). Even so, unless the keys are distributed pretty unevenly, I wouldn't expect any huge improvement -- a perfect hash function is often (usually?) fairly complex.
If your input is sorted, you should try just a vector and a binary search (i.e.,
lower_bound()
). This might prove adaquate (it is also O(log n)). Depending on the distribution of your keys and the hash function used, a hash_map might work as well. I think this istr1::unordered_map
in gcc.