使用C++简单实现LRU缓存

使用C++简单实现LRU缓存


  昨天碰巧复习到了LRU缓存机制,也就是最近最少使用,在好奇心的驱动下我想到了用C++简单的实现一下,本来原理也挺简单的,写出来并不难,但由于对C++的基础把握不劳,结果排查了好长时间的bug。现在总结一下,希望能帮到别的小可爱。

首先是malloc:

  可能是C语言写习惯的,导致在C++进行内存分配的时候我也习惯用malloc,C和C++混用很可能就会出bug!
  在给struct LRUCache{};初始化时,由于指针会指向一块内存,指针本身也是要占用内存的,但指针指向的地方还没有分配内存呢,所以就需要又malloc 。我就是少malloc,结果程序老是运行不下去。。。

接着是map:

  我在struct LRUCache{};放了一个map<int, ListNode*> mp; mp用来存索引key和映射ListNode的指针.
  如果是用malloc初始化struct LRUCache{};那么又需要给map初始化,因为malloc只是申请一块内存,不会初始化map对应的内存。使用new 除了malloc内存,还会调用默认的构造初始化函数 ,所以最好还是用new比较省事。

malloc和new的区别:

(1)malloc与free是C++/C语言的标准库函数,new/delete是C++的运算符。
(2)new操作符从自由存储区(free store)上为对象动态分配内存空间,而malloc函数从堆上动态分配内存。
(3)使用new操作符申请内存分配时无须指定内存块的大小,编译器会根据类型信息自行计算,而malloc则需要显式地指出所需内存的尺寸。

  下面是代码,用了get(key)获取数据,put(key, value)存放数据。代码还没有完善,只是简单的实现,后面有时间会进行修改。(其实可以把struct LRUCache{};换成一个类,这样比较好。。。)
代码如下:

#include<iostream>
#include<stdlib.h> 
#include<map>
using namespace std;
struct ListNode{
	int key;
	int value;
	ListNode *pre,*next;//指针 
	ListNode (int key1,int value1){
		key=key1;
		value=value1;
		pre=NULL;
		next=NULL;
	}
};
struct LRUCache{
	int Captacity;
	map<int,ListNode*> m;
	ListNode *head,*tail;//指针 
	LRUCache(int captacity){
		Captacity= captacity;
		head=new ListNode(0,0); 
    	tail=new ListNode(0,0); 
	}
}*cache;
LRUCache *Constructor(int capacity){//初始化 
    LRUCache *cache=new LRUCache(capacity); 
	cache->head->next=cache->tail;
	cache->tail->pre=cache->head;
	return cache;
}
void removeNode(ListNode *node){    //移除 
	node->pre->next=node->next;
	node->next->pre=node->pre;
}
void addNode(ListNode *node){//添加元素到表头 
	node->next=cache->head->next;
	node->pre=cache->head;
	cache->head->next->pre=node;
	cache->head->next=node;
} 
void moveToHead(ListNode *node) {//移动到表头 
	removeNode(node); 
	addNode(node);
}
void popTail(){ //删除表尾 
	ListNode *res = cache->tail->pre;
	removeNode(res);
}
int get(int key){
	map<int,ListNode*>::iterator it=cache->m.find(key);
	if(it==cache->m.end()){//如果不存在 
		return -1;
	}
	ListNode *node = it->second;
	moveToHead(node);
	return node->value; 
}
void put(int key,int value){
	map<int,ListNode*>::iterator it=cache->m.find(key);
	if(it==cache->m.end()){//如果不存在 
		ListNode * NewNode = new ListNode(key,value);
		if(cache->m.size()==cache->Captacity){//如果满了 
			cache->m.erase(cache->tail->pre->key);//从map表中删除最后一个 
			cache->m[key]= NewNode;//从map表中加上新的 
			popTail();//删除 cache最后一个 
			addNode(NewNode);//把新的移到最前面 
		}else{
			cache->m[key]= NewNode;
			addNode(NewNode);
		}
	} else{//如果存在 
		it->second->value=value;//更新值 
		moveToHead(it->second);
	}
}
int main(){
	cache=Constructor(3);
	put(1,1);
	put(2,2);
	put(3,3);
	//顺序:3,2,1 
	cout<<get(1)<<endl;//1
	// 顺序:1,3,2 
	put(4,4);
	// 顺序:4,1,3 
	cout<<get(2)<<endl;//-1
	put(5,5);
	// 顺序:5,4,1
	cout<<get(3)<<endl;//-1
	cout<<get(4)<<endl;//4
	cout<<get(5)<<endl;//5
	return 0;
} 

  如果哪里写得不够好或者是大家有不同看法的欢迎评论!我会看情况进行更改。

发布了9 篇原创文章 · 获赞 4 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/Milan_1in/article/details/105178483