Untitled

#include <queue>
#include <vector>
#include <set>
#include <iostream>
#define MAX_N 100
using namespace std;

int map[MAX_N + 1][MAX_N + 1];
int visit[MAX_N + 1][MAX_N + 1];
int attend[MAX_N + 1][MAX_N + 1];

struct cmp {
    bool operator()(pair<int, int> a, pair<int, int> b) {
        if (map[a.first][a.second] == map[b.first][b.second]) {
            if (a.first == b.first) {
                return a.second > b.second;
            }
            return a.first > b.first;
        }
        return map[a.first][a.second] < map[b.first][b.second];
    }
};

priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> pq;
set<pair<int, int>> medicine[3];

int dx[4] = {-1, 0, 1, 0};
int dy[4] = {0, 1, 0, -1};
int n;
int step;

void init(int N, int mDish[MAX_N][MAX_N]) {
    n = N;
    step = 1;
    for (int i = 1; i <= N; i++) {
        for (int j = 1; j <= N; j++) {
            map[i][j] = mDish[i-1][j-1]; // Chỉnh lại để truy cập đúng chỉ số
            visit[i][j] = 0;
            attend[i][j] = 0;
        }
    }
}

int dropMedicine(int mTarget, int mRow, int mCol, int mEnergy) {
    int e = mEnergy;

    // Bắt đầu rải thuốc tại ô ban đầu
    if (visit[mRow][mCol] == 0) {
        e -= map[mRow][mCol];  // Trừ năng lượng tại ô hiện tại
        visit[mRow][mCol] = step++;
        attend[mRow][mCol] = 1;
        medicine[mTarget].insert(make_pair(mRow, mCol));
        cout << "Map[" << mRow << "][" << mCol << "] : " << map[mRow][mCol] << endl;
    }

    // Đẩy các ô lân cận vào priority_queue
    for (int k = 0; k < 4; k++) {
        int nx = mRow + dx[k];
        int ny = mCol + dy[k];
        if (nx >= 1 && nx <= n && ny >= 1 && ny <= n && !attend[nx][ny]) {
            attend[nx][ny] = 1;
            pq.push(make_pair(nx, ny));
        }
    }

    // Xử lý việc rải thuốc dựa trên năng lượng còn lại và priority_queue
    while (e > 0 && !pq.empty()) {
        auto topPair = pq.top();
        pq.pop();
        int r = topPair.first;
        int c = topPair.second;

        if (!visit[r][c]) {  // Kiểm tra xem ô này có bị thăm hay chưa
            visit[r][c] = step++;
            medicine[mTarget].insert(make_pair(r, c));
            e -= map[r][c];  // Trừ năng lượng tại ô hiện tại

            // Nếu năng lượng còn, tiếp tục thêm các ô lân cận vào hàng đợi
            if (e > 0) {
                for (int k = 0; k < 4; k++) {
                    int nx = r + dx[k];
                    int ny = c + dy[k];
                    if (nx >= 1 && nx <= n && ny >= 1 && ny <= n && !attend[nx][ny]) {
                        attend[nx][ny] = 1;
                        pq.push(make_pair(nx, ny));
                    }
                }
            }
        }
    }

    // Hiển thị kết quả mảng visit
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            cout << visit[i][j] << " ";
        }
        cout << endl;
    }

    // Hiển thị kết quả mảng map
    cout << "--------------" << endl;
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            cout << map[i][j] << " ";
        }
        cout << endl;
    }

    return medicine[mTarget].size();
}

int cleanBacteria(int mRow, int mCol) {
    // Tạm thời trả về -1
    return -1;
}