Guest User

Untitled

a guest
Feb 17th, 2018
90
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.58 KB | None | 0 0
  1. #ifndef _LRUCACHE_HPP_INCLUDED_
  2. #define _LRUCACHE_HPP_INCLUDED_
  3.  
  4. #include <unordered_map>
  5. #include <list>
  6. #include <cstddef>
  7. #include <stdexcept>
  8.  
  9. namespace cache {
  10.  
  11. template<typename key_t, typename value_t>
  12. class lru_cache {
  13. public:
  14. typedef typename std::pair<key_t, value_t> key_value_pair_t;
  15. typedef typename std::list<key_value_pair_t>::iterator list_iterator_t;
  16.  
  17. lru_cache(size_t max_size) :
  18. _max_size(max_size) {
  19. }
  20.  
  21. void put(const key_t& key, const value_t& value) {
  22. auto it = _cache_items_map.find(key);
  23. _cache_items_list.push_front(key_value_pair_t(key, value));
  24. if (it != _cache_items_map.end()) {
  25. _cache_items_list.erase(it->second);
  26. _cache_items_map.erase(it);
  27. }
  28. _cache_items_map[key] = _cache_items_list.begin();
  29.  
  30. if (_cache_items_map.size() > _max_size) {
  31. auto last = _cache_items_list.end();
  32. last--;
  33. _cache_items_map.erase(last->first);
  34. _cache_items_list.pop_back();
  35. }
  36. }
  37.  
  38. const value_t& get(const key_t& key) {
  39. auto it = _cache_items_map.find(key);
  40. if (it == _cache_items_map.end()) {
  41. throw std::range_error("There is no such key in cache");
  42. } else {
  43. _cache_items_list.splice(_cache_items_list.begin(), _cache_items_list, it->second);
  44. return it->second->second;
  45. }
  46. }
  47.  
  48. bool exists(const key_t& key) const {
  49. return _cache_items_map.find(key) != _cache_items_map.end();
  50. }
  51.  
  52. size_t size() const {
  53. return _cache_items_map.size();
  54. }
  55.  
  56. private:
  57. std::list<key_value_pair_t> _cache_items_list;
  58. std::unordered_map<key_t, list_iterator_t> _cache_items_map;
  59. size_t _max_size;
  60. };
  61.  
  62. } // namespace cache
  63.  
  64. #endif /* _LRUCACHE_HPP_INCLUDED_ */
Add Comment
Please, Sign In to add comment