380. Insert Delete GetRandom O(1).cpp

#include<bits/stdc++.h>
using namespace std;
 

class myStructure
{

    vector <int> arr;

    unordered_map <int, int> Map;
 
    public:
   
    void add(int x)
    {

        if(Map.find(x) != Map.end())
            return;
        
        int index = arr.size();
        arr.push_back(x);
             
       
        Map.insert(std::pair<int,int>(x, index));
    }
         
 
    void remove(int x)
    {
   
        if(Map.find(x) == Map.end())
            return;
     
        int index = Map.at(x);
        Map.erase(x);
        
        int last = arr.size() - 1;
        swap(arr[index], arr[last]);
        arr.pop_back();
             
        Map.at(arr[index]) = index;
    }
         
   
    int search(int x)
    {
        if(Map.find(x) != Map.end())
        return Map.at(x);
        return -1;
    }
        
    int getRandom()
    {
        srand (time(NULL));
        int random_index = rand() % arr.size();
      
        return arr.at(random_index);
    }     
};
 
int main()
{
    myStructure ds;
    ds.add(10);
    ds.add(900);
    ds.add(79);
    ds.add(490);
    cout << ds.search(900) << endl;
    ds.remove(900);
    ds.add(50);
    cout << ds.search(50) << endl;
    cout << ds.getRandom() << endl;
}