MemPool

腾讯笔试题，设计内存池，alloc和free都是O(1)。

和LRUCache类似，这里用了一个list表示可用的空间，用一个map来记录这块内存是否已分配，这样free的时候才可能O(1)。

 1 class MemPool {

 2     public:

 3         void init(int unitSize, int maxUnitNum) {

 4             long long size = unitSize * maxUnitNum;

 5             buffer = new char[size];

 6             memset(buffer, 0, sizeof(char) * size);

 7             for (int i = 0; i < size; ++i) {

 8                 available.push_back(buffer + i);

 9             }

10         }

11 

12         ~MemPool() {

13             if (buffer) {

14                 delete[] buffer;

15             }

16         }

17         // O(1)

18         void* alloc() {

19             if (available.empty()) return NULL;

20             char* ans = available.front();

21             available.pop_front();

22             allocated[ans] = 1;

23             return ans;

24         }

25 

26         // O(1)

27         void free(void *pUnit) {

28             char* tmp = (char*)pUnit;

29             if (allocated.find(tmp) == allocated.end()

30                     || allocated[tmp] == 0) return;

31             available.push_back(tmp);

32             allocated[tmp] = 0;

33         }

34     private:

35         unordered_map<char*, int> allocated;

36         list<char*> available;

37         char* buffer;

38 };