Untitled

#include <unordered_map>
#include <set>
#include <vector>
#include <algorithm>

using namespace std;

struct Product {
    int rank;
    int score;
    bool deleted;
};

const int MAX_CATEGORIES = 6;
const int MAX_SCORE = 101;

unordered_map<int, Product> global[MAX_CATEGORIES];
unordered_map<int, int> global_all;

// Set to maintain order of products by rank and ID
set<pair<int, int>> prod_data[MAX_CATEGORIES]; // set<rank, GoodsID>

int rank_map[MAX_SCORE];

void init() {
    for (int i = 0; i < MAX_SCORE; ++i) {
        rank_map[i] = i <= 19 ? 5 : i <= 39 ? 4 : i <= 59 ? 3 : i <= 79 ? 2 : 1;
    }

    for (int i = 1; i < MAX_CATEGORIES; ++i) {
        global[i].clear();
        prod_data[i].clear();
    }
    global_all.clear();
}

void add(int mGoodsID, int mCategory, int mScore) {
    Product temp{rank_map[mScore], mScore, false};
    global[mCategory][mGoodsID] = temp;
    global_all[mGoodsID] = mCategory;
    prod_data[mCategory].emplace(temp.rank, mGoodsID);
}

void remove(int mGoodsID) {
    global[global_all[mGoodsID]][mGoodsID].deleted = true;
}

void updateRank(int mGoodsID, int newScore) {
    int mCategory = global_all[mGoodsID];
    Product& prod = global[mCategory][mGoodsID];
    int oldRank = prod.rank;
    prod.score = newScore;
    prod.rank = rank_map[newScore];

    if (oldRank != prod.rank) {
        prod_data[mCategory].erase({oldRank, mGoodsID});
        prod_data[mCategory].emplace(prod.rank, mGoodsID);
    }
}

void purchase(int mGoodsID) {
    int newScore = min(global[global_all[mGoodsID]][mGoodsID].score + 5, 100);
    updateRank(mGoodsID, newScore);
}

void takeBack(int mGoodsID) {
    int newScore = max(global[global_all[mGoodsID]][mGoodsID].score - 10, 0);
    updateRank(mGoodsID, newScore);
}

void changeScore(int mCategory, int mChangeScore) {
    for (auto& [id, prod] : global[mCategory]) {
        if (!prod.deleted) {
            int newScore = clamp(prod.score + mChangeScore, 0, 100);
            updateRank(id, newScore);
        }
    }
}

int getTopRank(int mCategory) {
    if (mCategory == 0) {
        pair<int, int> ans{5, 0};
        for (int i = 1; i < MAX_CATEGORIES; ++i) {
            for (const auto& [rank, id] : prod_data[i]) {
                if (!global[i][id].deleted) {
                    if (rank < ans.first || (rank == ans.first && id > ans.second)) {
                        ans = {rank, id};
                    }
                }
            }
        }
        return ans.second;
    } else {
        for (const auto& [rank, id] : prod_data[mCategory]) {
            if (!global[mCategory][id].deleted) {
                return id;
            }
        }
    }
    return 0;
}