Submission #878964

#	Time	Username	Problem	Language	Result	Execution time	Memory
878964		ArashJafariyan	Meteors (POI11_met)	C++17	100 / 100	1519 ms	65536 KiB

met

#include <bits/stdc++.h>
using namespace std;
 
#define pb push_back
#define ll long long

const int N = 3e5 + 1;
 
int n, m, k, p[N], lq[N], rq[N], aq[N], l[N], mid[N], r[N];
ll sum[N];
vector<int> own[N], mids[N];
 
bool update() {
    for (int i = 1; i <= k + 1; ++i) {
        mids[i].clear();
    }
    for (int i = 1; i <= m; ++i) {
        sum[i] = 0;
    } 
 
    bool ret = false;
    for (int i = 1; i <= n; ++i) {
        mid[i] = (l[i] + r[i]) / 2;
        mids[mid[i]].pb(i);
        if (r[i] - l[i] > 1) {
            ret = true;
        }
    }
    return ret;
}
 
void modify(int id, int val) {
    for (; id <= m; id += id & -id) {
        sum[id] += val;
    }
}
 
ll get(int id) {
    ll ret = 0;
    for (; id; id -= id & -id) {
        ret += sum[id];
    }
    return ret;
}
 
int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
 
    cin >> n >> m;
    for (int i = 1; i <= m; ++i) {
        int o;
        cin >> o;
        own[o].pb(i);
    }
    for (int i = 1; i <= n; ++i) {
        cin >> p[i];
    }
    cin >> k;
    for (int i = 1; i <= k; ++i) {
       cin >> lq[i] >> rq[i] >> aq[i];
    }
 
    lq[k + 1] = 1;
    rq[k + 1] = m;
    aq[k + 1] = (int) 1e9;
    for (int i = 1; i <= n; ++i) {
        l[i] = 0;
        r[i] = k + 1;
    }
    while (update() == true) {
        for (int i = 1; i <= k + 1; ++i) {
            if (lq[i] > rq[i]) {
                modify(1, aq[i]);
                modify(rq[i] + 1, -aq[i]);
                modify(lq[i], aq[i]);
            }
            else {
                modify(lq[i], aq[i]);
                modify(rq[i] + 1, -aq[i]);
            }
           
            for (int idx : mids[i]) {
                ll s = 0;
                for (int o : own[idx]) {
                    s += get(o);
                    if (s >= p[idx]) {
                        break;
                    }
                }
                if (s >= p[idx]) {
                    r[idx] = i;
                }
                else {
                    l[idx] = i;
                }
            }
        }
    }
 
    for (int i = 1; i <= n; ++i) {
        if (r[i] == k + 1) {
            cout << "NIE";
        }
        else {
            cout << r[i];
        }
        cout << '\n';
    }
 
    return 0;
}

• Subtask 1: 12 / 12
• Subtask 2: 12 / 12
• Subtask 3: 12 / 12
• Subtask 4: 12 / 12
• Subtask 5: 13 / 13
• Subtask 6: 13 / 13
• Subtask 7: 13 / 13
• Subtask 8: 13 / 13