oj.uz

Submission #1042956

# Submission time Handle Problem Language Result Execution time Memory
1042956 2024-08-03T15:39:00 Z armashka Meteors (POI11_met) C++17
74 / 100
 3281 ms 59548 KB 
#include<bits/stdc++.h>
#include <unordered_map>

#pragma GCC target("avx2")
#pragma GCC optimize("Ofast")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")

using namespace std;

#define fast ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define file(s) freopen(s".in", "r", stdin);freopen(s".out", "w", stdout);
#define all(v) v.begin(),v.end()
#define rall(v) v.rbegin(),v.rend()
#define pb push_back
#define pf push_front
#define ppb pop_back
#define ft first
#define sd second
#define ll long long
#define ull unsigned ll
#define pii pair<int,int>
#define pll pair<ll,ll>
const int N = 3e5 + 10;
const int M = 1e6 + 5;
const int B = 316;
const ll msize = 2;
const ll mod1 = 1e9 + 7;
const ll mod2 = 998244353;
const long double Phi = acos(-1);
const long long inf = 2e18;
const vector <pair<int, int>> dir = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};

ll binmul(ll x, ll ti, ll m);
ll binpow(ll x, ll ti, ll m);

struct meteor {
    ll l, r, x;
} a[N];

struct Government {
    ll l, r, ans, id, need;
};

ll n, m, k;
ll t[N * 4], fl[N * 4];
vector <ll> gov[N];

void push(ll v, ll tl, ll tr) {
    t[v] += fl[v] * (tr - tl + 1);
    if (tl < tr) {
        fl[v + v] += fl[v];
        fl[v + v + 1] += fl[v];
    }
    fl[v] = 0;
}

void upd(ll l, ll r, ll x, ll v = 1, ll tl = 1, ll tr = m) {
    push(v, tl, tr);
    if (r < tl || tr < l) return;
    if (l <= tl && tr <= r) {
        fl[v] += x;
        push(v, tl, tr);
        return;
    }
    ll tm = (tl + tr) / 2;
    upd(l, r, x, v + v, tl, tm);
    upd(l, r, x, v + v + 1, tm + 1, tr);
    t[v] = t[v + v] + t[v + v + 1];
}

ll get(ll pos, ll v = 1, ll tl = 1, ll tr = m) {
    push(v, tl, tr);
    if (tl == tr) return t[v];
    int tm = (tl + tr) / 2;
    if (pos <= tm) return get(pos, v + v, tl, tm);
    else return get(pos, v + v + 1, tm + 1, tr);
}

const void solve() {
    cin >> n >> m;
    for (ll i = 1; i <= m; ++ i) {
        int id; cin >> id;
        gov[id].pb(i);
    }

    vector <Government> p;
    p.pb({0, 0, 0, 0});
    for (ll i = 1; i <= n; ++ i) {
        ll x; cin >> x;
        p.pb({0, 0, -1, i, x});
    }

    cin >> k;
    for (ll i = 1; i <= k; ++ i) {
        cin >> a[i].l >> a[i].r >> a[i].x;
    }

    for (ll i = 1; i <= n; ++ i) p[i].l = 1, p[i].r = k;

    for (ll i = 1; i <= 20; ++ i) {
        vector <ll> g[k + 1];
        for (ll j = 1; j <= m * 4; ++ j) t[j] = fl[j] = 0;

        for (ll j = 1; j <= n; ++ j) {
            int mid = (p[j].l + p[j].r) / 2;
            g[mid].pb(p[j].id);
        }

        for (ll j = 1; j <= k; ++ j) {
            if (a[j].l <= a[j].r) {
                upd(a[j].l, a[j].r, a[j].x);
            } else {
                upd(a[j].l, m, a[j].x);
                upd(1, a[j].r, a[j].x);
            }

            for (auto id : g[j]) {
                ll sum = 0;
                for (auto location : gov[id]) {
                    sum += get(location);
                }

                ll mid = (p[id].l + p[id].r) / 2;
                if (sum >= p[id].need) {
                    p[id].r = mid - 1;
                    p[id].ans = mid;
                } else {
                    p[id].l = mid + 1;
                }
            }

        }
    }

    for (ll i = 1; i <= n; ++ i) {
        if (p[i].ans == -1) cout << "NIE\n";
        else cout << p[i].ans << "\n";
    }
}

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    srand(time(NULL));

    // file("promote");

    // freopen("input.txt", "r", stdin);
    // freopen("output.txt", "w", stdout);

    int tt = 1;
    // cin >> tt;
    for (int i = 1; i <= tt; ++ i) {
        // cout << "Caso #" << i << "\n";
        solve();
    }

    #ifndef ONLINE_JUDGE
    cerr << "\n" << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
    #endif

    return 0;
}


// Template functions
ll binmul(ll x, ll ti, ll m) { ll res = 0;while (ti){if(ti & 1)res += x;x += x;ti >>= 1; x %= m; res %= m;} return res;}
ll binpow(ll x, ll ti, ll m) { ll res = 1;while (ti){if(ti & 1)(res *= x)%=m;(x*=x)%=m;ti >>= 1; x %= m; res %= m;} return res;}

Subtask #1
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 12888 KB Output is correct
2 Correct 7 ms 12888 KB Output is correct
3 Correct 9 ms 12888 KB Output is correct

Subtask #2
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 13016 KB Output is correct
2 Correct 6 ms 13044 KB Output is correct
3 Correct 7 ms 12948 KB Output is correct

Subtask #3
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 388 ms 22068 KB Output is correct
2 Correct 426 ms 23808 KB Output is correct
3 Correct 427 ms 22848 KB Output is correct

Subtask #4
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 433 ms 22612 KB Output is correct
2 Correct 423 ms 22628 KB Output is correct
3 Correct 430 ms 23760 KB Output is correct
4 Correct 100 ms 19280 KB Output is correct

Subtask #5
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 202 ms 21908 KB Output is correct
2 Correct 339 ms 23732 KB Output is correct
3 Correct 165 ms 16768 KB Output is correct
4 Correct 426 ms 23248 KB Output is correct

Subtask #6
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 384 ms 21720 KB Output is correct
2 Correct 386 ms 22760 KB Output is correct
3 Correct 382 ms 21816 KB Output is correct
4 Correct 457 ms 24116 KB Output is correct

Subtask #7
0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 3281 ms 59548 KB Output is correct
2 Incorrect 2401 ms 50532 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #8
0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 3279 ms 59392 KB Output is correct
2 Incorrect 1323 ms 48744 KB Output isn't correct
3 Halted 0 ms 0 KB -