Abstract Classes - Polymorphism in C++

Solution in C++

#include <iostream>

#include <vector>

#include <map>

#include <string>

#include <algorithm>

#include <set>

#include <cassert>

using namespace std;

struct Node{

   Node* next;

   Node* prev;

   int value;

   int key;

   Node(Node* p, Node* n, int k, int val):prev(p),next(n),key(k),value(val){};

   Node(int k, int val):prev(NULL),next(NULL),key(k),value(val){};

};

class Cache{

   

   protected: 

   map<int,Node*> mp; //map the key to the node in the linked list

   int cp;  //capacity

   Node* tail; // double linked list tail pointer

   Node* head; // double linked list head pointer

   virtual void set(int, int) = 0; //set function

   virtual int get(int) = 0; //get function

};

#include <iostream>...

class LRUCache : public Cache{

public:

    LRUCache(int capacity) {

        cp = capacity;

        tail = NULL;

        head = NULL;

    }

    void set(int key, int val) {

        if(mp.find(key) != mp.end())

            mp[key]->value = val;

        else {

            mp[key] = new Node(NULL, head, key, val);

            if(head) head->prev = mp[key];

            head = mp[key];

            if(!tail) tail = head;

            if(mp.size() > cp) {

                mp.erase(tail->key);

                tail = tail->prev;

                delete tail->next;

                tail->next = NULL;

            }

        }

    }

    int get(int key) {

        if(mp.find(key) != mp.end()) {

            if(!mp[key]->next)

                tail = mp[key]->prev;

            head = mp[key];

            while(mp[key]->prev) {

                Node* temp = mp[key]->prev;

                mp[key]->prev = mp[key]->prev->prev;

                if(mp[key]->prev)

                    mp[key]->prev->next = mp[key];

                if(mp[key]->next)

                    mp[key]->next->prev = temp;

                temp->next = mp[key]->next;

                temp->prev = mp[key];

                mp[key]->next = temp;

            }

            return mp[key]->value;

        }

        else

            return -1;

    }

};

int main() {

   int n, capacity,i;

   cin >> n >> capacity;

   LRUCache l(capacity);

   for(i=0;i<n;i++) {

      string command;

      cin >> command;

      if(command == "get") {

         int key;

         cin >> key;

         cout << l.get(key) << endl;

      } 

      else if(command == "set") {

         int key, value;

         cin >> key >> value;

         l.set(key,value);

      }

   }

   return 0;

}