LeetCode 1912. 设计电影租借系统（map+set）

示例 1：
输入：
["MovieRentingSystem", "search", "rent", "rent", "report", "drop", "search"]
[[3, [[0, 1, 5], [0, 2, 6], [0, 3, 7], [1, 1, 4], [1, 2, 7], [2, 1, 5]]], [1], [0, 1], [1, 2], [], [1, 2], [2]]
输出：
[null, [1, 0, 2], null, null, [[0, 1], [1, 2]], null, [0, 1]]

解释：
MovieRentingSystem movieRentingSystem = new MovieRentingSystem(3, [[0, 1, 5], [0, 2, 6], [0, 3, 7], [1, 1, 4], [1, 2, 7], [2, 1, 5]]);
movieRentingSystem.search(1);  // 返回 [1, 0, 2] ，商店 1，0 和 2 有未借出的 ID 为 1 的电影。商店 1 最便宜，商店 0 和 2 价格相同，所以按商店编号排序。
movieRentingSystem.rent(0, 1); // 从商店 0 借出电影 1 。现在商店 0 未借出电影编号为 [2,3] 。
movieRentingSystem.rent(1, 2); // 从商店 1 借出电影 2 。现在商店 1 未借出的电影编号为 [1] 。
movieRentingSystem.report();   // 返回 [[0, 1], [1, 2]] 。商店 0 借出的电影 1 最便宜，然后是商店 1 借出的电影 2 。
movieRentingSystem.drop(1, 2); // 在商店 1 返还电影 2 。现在商店 1 未借出的电影编号为 [1,2] 。
movieRentingSystem.search(2);  // 返回 [0, 1] 。商店 0 和 1 有未借出的 ID 为 2 的电影。商店 0 最便宜，然后是商店 1 。
 
提示：
1 <= n <= 3 * 10^5
1 <= entries.length <= 10^5
0 <= shopi < n
1 <= moviei, pricei <= 10^4
每个商店 至多 有一份电影 moviei 的拷贝。
search，rent，drop 和 report 的调用 总共 不超过 10^5 次。

class MovieRentingSystem {
    unordered_map<int, set<pair<int,int>>> unborrowed; //movie : <price, shop>
    set<vector<int>> borrowed; // <price, shop, movie>
    unordered_map<long long, int> m; // shop*k+movie, price, 根据 shop，movie，获取其 price
    long long k = 10001;
public:
    MovieRentingSystem(int n, vector<vector<int>>& entries) {
        for(auto& e : entries)
        {
            unborrowed[e[1]].insert({e[2], e[0]});
            m[e[0]*k+e[1]] = e[2];
        }
    }
    
    vector<int> search(int movie) {
        if(unborrowed.find(movie) == unborrowed.end())
            return {};
        auto it = unborrowed[movie].begin();
        int i = 0;
        vector<int> ans;
        while(i < 5 && it != unborrowed[movie].end()) // 未借出电影的最便宜的5家店
        {
            ans.push_back((*it).second);
            i++;
            it++;
        }
        return ans;
    }
    
    void rent(int shop, int movie) {
        int price = m[shop*k+movie];
        borrowed.insert({price, shop, movie});//借出
        unborrowed[movie].erase({price, shop});
        if(unborrowed[movie].empty())
            unborrowed.erase(movie);
    }
    
    void drop(int shop, int movie) {
        int price = m[shop*k+movie];
        borrowed.erase({price, shop, movie});
        unborrowed[movie].insert({price, shop});//归还
    }
    
    vector<vector<int>> report() {
        auto it = borrowed.begin();
        int i = 0;
        vector<vector<int>> ans;
        while(i < 5 && it != borrowed.end())//借出里面最便宜的5本
        {
            ans.push_back({(*it)[1], (*it)[2]});
            i++;
            it++;
        }
        return ans;
    }
};