Untitled

#include <iostream>
#include <set>
#include <algorithm>
using namespace std;

struct res {
    int mCost, mContent;
};

struct Mineral {
    int cost, mineID, content;
};

struct comp {
    bool operator()(Mineral m1, Mineral m2) const {
        return m1.cost < m2.cost;
    }
};

set<Mineral, comp> mine_comb[2][3]; // Mảng set để lưu các viên đá theo mỏ và loại
int shipfee;

void init(int mShipFee) {
    for (int i = 0; i < 3; i++) {
        for (int j = 0; j < 3; j++)
            mine_comb[i][j].clear(); // Xóa tất cả các set
    }
    shipfee = mShipFee; // Thiết lập phí vận chuyển
}

int gather(int mMineId, int mType, int mCost, int mContent) {
    Mineral mid;
    mid.content = mContent;
    mid.cost = mCost;
    mid.mineID = mMineId;
    mine_comb[mMineId][mType].insert(mid); // Thêm viên đá vào mỏ cụ thể
    mine_comb[2][mType].insert(mid); // Thêm viên đá vào hàng tổng hợp (row 2)
    return mine_comb[mMineId][mType].size();
}

res mix(int mBudget) {
    int row = -1;
    res res;
    res.mContent = 0;
    res.mCost = 0;
    int price = 0;
    int content = -1;
    Mineral del[3]; // Lưu các viên đá sẽ bị xóa

    for (int i = 0; i <