Merge pull request SjxSubham#69 from shafaq16/patch-4

1912. Design Movie Rental System.cpp

Author: SjxSubham

1912. Design Movie Rental System.cpp

@@ -0,0 +1,53 @@
+using int2=pair<int,int>;           // (price, shop)
+using int3=tuple<int,int,int>;      // (price, shop, movie)
+
+class MovieRentingSystem {
+public:
+    unordered_map<uint64_t,int> Shop;      // (shop,movie)->price
+    unordered_map<int,set<int2>> Movie;    // movie->set(price,shop)
+    set<int3> rented;                      // rented->set (price,shop,movie)
+
+    static inline uint64_t key(int shop,int movie) {
+        return (uint64_t)shop<<32|movie;
+    }
+
+    MovieRentingSystem(int n, vector<vector<int>>& entries) {
+        for (auto &e : entries) {
+            int shop=e[0], movie=e[1], price=e[2];
+            Shop[key(shop,movie)]=price;
+            Movie[movie].insert({price, shop});
+        }
+    }
+
+    vector<int> search(int movie) {
+        vector<int> ans;
+        auto& S=Movie[movie];
+        int i=0;
+        for (auto it=S.begin(); it!=S.end() && i<5; it++, i++) {
+            ans.push_back(it->second); // shop
+        }
+        return ans;
+    }
+
+    void rent(int shop,int movie) {
+        int price=Shop[key(shop,movie)];
+        Movie[movie].erase({price, shop});
+        rented.insert({price, shop, movie});
+    }
+
+    void drop(int shop,int movie) {
+        int price=Shop[key(shop,movie)];
+        Movie[movie].insert({price, shop});
+        rented.erase({price, shop, movie});
+    }
+
+    vector<vector<int>> report() {
+        vector<vector<int>> ans;
+        int i=0;
+        for (auto it=rented.begin(); it!=rented.end() && i<5; it++, i++) {
+            auto [price, shop, movie]=*it;
+            ans.push_back({shop, movie});
+        }
+        return ans;
+    }
+};