MemPool

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

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

class MemPool {
    public:
        void init(int unitSize, int maxUnitNum) {
            long long size = unitSize * maxUnitNum;
            buffer = new char[size];
            memset(buffer, 0, sizeof(char) * size);
            for (int i = 0; i < size; ++i) {
                available.push_back(buffer + i);
            }
        }

        ~MemPool() {
            if (buffer) {
                delete[] buffer;
            }
        }
        // O(1)
        void* alloc() {
            if (available.empty()) return NULL;
            char* ans = available.front();
            available.pop_front();
            allocated[ans] = 1;
            return ans;
        }

        // O(1)
        void free(void *pUnit) {
            char* tmp = (char*)pUnit;
            if (allocated.find(tmp) == allocated.end()
                    || allocated[tmp] == 0) return;
            available.push_back(tmp);
            allocated[tmp] = 0;
        }
    private:
        unordered_map<char*, int> allocated;
        list<char*> available;
        char* buffer;
};