LeetCode5783. 设计电影租借系统（set，模拟）

Posted 2021-07-23 live4m

题意：

解法：

设三元组为(x,z,y),
1.对每种电影开一个set,存当前电影未借出的(价格,商店),即(z,x).
2.对每种商店开一个set,存当前商店未借出的(电影,价格),即(y,z).
3.对每种商店开一个set,存当前商店已借出的(电影,价格),即(y,z).
4.开一个set,存已借出的所有电影(x,y,z).

对于search,直接用情况1中的set查即可.
对于rent和drop,维护一下各个set即可.
对于report,用情况4中的set查即可.

code：

const int maxm=3e5+5;
#define PI pair<int,int>
#define PII pair<int,pair<int,int> >
class MovieRentingSystem {
public:
    set<PI>zx[maxm];//存每种电影现在的zx
    set<PI>yz[maxm];//存每种商店现在的电影
    //
    set<PI>yz1[maxm];//存每种商店借出的电影
    set<PII>zxy;//已经借出的所有电影
    //
    MovieRentingSystem(int n, vector<vector<int>>& entries) {
        for(int i=1;i<=1e4;i++)zx[i].clear();
        for(int i=1;i<=n;i++)yz[i].clear();
        for(int i=1;i<=n;i++)yz1[i].clear();
        zxy.clear();
        for(auto i:entries){
            int x=i[0],y=i[1],z=i[2];
            zx[y].insert({z,x});
            yz[x].insert({y,z});
        }
    }

    vector<int> search(int vid) {
        vector<int>ans;
        auto it=zx[vid].begin();
        int cnt=0;
        while(it!=zx[vid].end()&&cnt<5){
            ans.push_back(it->second);
            it++;
            cnt++;
        }
        return ans;
    }
    
    void rent(int sid, int vid) {
        auto it=yz[sid].lower_bound({vid,0});
        int x=sid,y=vid,z=it->second;
        yz[x].erase({y,z});
        zx[y].erase({z,x});
        yz1[x].insert({y,z});
        zxy.insert({z,{x,y}});
    }
    
    void drop(int sid, int vid) {
        auto it=yz1[sid].lower_bound({vid,0});
        int x=sid,y=vid,z=it->second;
        yz[x].insert({y,z});
        zx[y].insert({z,x});
        yz1[x].erase({y,z});
        zxy.erase({z,{x,y}});
    }
    
    vector<vector<int>> report() {
        vector<vector<int> >ans;
        auto it=zxy.begin();
        int cnt=0;
        while(it!=zxy.end()&&cnt<5){
            ans.push_back({it->second.first,it->second.second});
            it++;
            cnt++;
        }
        return ans;
    }
};

/**
 * Your MovieRentingSystem object will be instantiated and called as such:
 * MovieRentingSystem* obj = new MovieRentingSystem(n, entries);
 * vector<int> param_1 = obj->search(movie);
 * obj->rent(shop,movie);
 * obj->drop(shop,movie);
 * vector<vector<int>> param_4 = obj->report();
 */

---