oj.uz

답안 #496463

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
496463 2021-12-21T09:09:04 Z ergagan Meteors (POI11_met) C++17
74 / 100
 6000 ms 50668 KB 
//я так много думал, что опять попал
#include <bits/stdc++.h>
#define all(x) x.begin(),x.end()
#define pb push_back
#define ppb pop_back
#define pf push_front
#define ppf pop_front
#define f first
#define s second
#define left(v) v + v
#define right(v) v + v + 1
#define ub upper_bound
#define lb lower_bound

#pragma comment(linker, "/stack:200000000")
#pragma GCC optimize("Ofast")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")

using namespace std;
typedef int ll;
typedef long long i64;

//17 SEVENTEEN
const long double Pi = acos(-1.0);
const ll dx[] = {0,0,1,-1};
const ll dy[] = {1,-1,0,0};
const ll N = (ll) 3e5 + 17;
const ll M = (ll) 5e3 + 69;
const ll inf = (ll) 1e14 + 3;
const ll mod = (ll) 1e9 + 7;
ll sq(ll x) { return x * x; }

ll zxc = 1, n, m, q, a[N];
i64 t[4 * N], uv[4 * N];
ll l[N], r[N], x[N];
ll p[N], cur[N], lw[N], hg[N];

vector<ll> v[N], vq[N];

void recalc(ll v) {
    t[v] = t[left(v)] + t[right(v)];
}

void push(ll v, ll tl, ll tr) {
    if (!uv[v]) return;
    if (tl < tr) {
        uv[left(v)] += uv[v];
        uv[right(v)] += uv[v];
    }
    t[v] += uv[v] * 1ll * (tr - tl + 1);
    uv[v] = 0;
}

ll get(ll v, ll tl, ll tr, ll l, ll r) {
    push(v, tl, tr);
    if(l <= tl && tr <= r) return t[v];
    if(tr < l || r < tl) return 0;
    ll tm = (tl + tr) / 2;
    return get(left(v), tl, tm, l, r) + get(right(v), tm + 1, tr, l, r);
}

void upd(ll v, ll tl, ll tr, ll ql, ll qr, ll val) {
    push(v, tl, tr);
    if (tr < ql || qr < tl) return;
    if (ql <= tl && tr <= qr) {
        uv[v] += val;
        push(v, tl, tr);
        return;
    }
    int mid = (tl + tr) / 2;
    upd(left(v), tl, mid, ql, qr, val);
    upd(right(v), mid + 1, tr, ql, qr, val);
    recalc(v);
}

void solve() {

    cin >> n >> m;

    for(ll i = 1; i <= m; i++) {
        cin >> p[i];
        v[p[i]].pb(i);
    }

    for(ll i = 1; i <= n; i++) {
        cin >> a[i];
    }

    cin >> q;

    for(ll i = 1; i <= n; i++)
        lw[i] = 1, hg[i] = q + 1;

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

    ll ok = 1;
    while(ok) {
        ok = 0;
//        cout << "OK\n";

        for(ll i = 1; i <= 4 * m; i++) t[i] = uv[i] = 0;

        for(ll i = 1; i <= n; i++) {
            if(lw[i] != hg[i])
                vq[(lw[i] + hg[i]) / 2].pb(i);
        }

        for(ll i = 1; i <= q; i++) {

            if(l[i] <= r[i])
                upd(1, 1, m, l[i], r[i], x[i]);

            else upd(1, 1, m, l[i], m, x[i]), upd(1, 1, m, 1, r[i], x[i]);

            while(!vq[i].empty()) {
                ok = 1;

                ll id = vq[i].back();
                vq[i].pop_back();

                i64 sum = 0;
                for(ll x : v[id]) {
                    sum += get(1, 1, m, x, x);
                }
                if(sum >= a[id]) hg[id] = i;
                else lw[id] = i + 1;
            }
        }
    }

    for(ll i = 1; i <= n; i++) {
        cout << (lw[i] <= q ? to_string(lw[i]) : "NIE") << "\n";
    }
}

int main(/*Уверенно*/) {
ios_base::sync_with_stdio(0);
    cin.tie(0);
/*
	freopen(".in", "r", stdin);
	freopen(".out", "w", stdout);
*/
//    cin >> zxc;
    while(zxc--) {
        solve();
    }
  	return 0;
}
// さよならさ いかなくちゃ

Compilation message

met.cpp:15: warning: ignoring '#pragma comment ' [-Wunknown-pragmas]
   15 | #pragma comment(linker, "/stack:200000000")
      |

Subtask #1
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 13 ms 14540 KB Output is correct
2 Correct 12 ms 14592 KB Output is correct
3 Correct 13 ms 14568 KB Output is correct

Subtask #2
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 12 ms 14560 KB Output is correct
2 Correct 11 ms 14472 KB Output is correct
3 Correct 12 ms 14620 KB Output is correct

Subtask #3
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 531 ms 19156 KB Output is correct
2 Correct 592 ms 21040 KB Output is correct
3 Correct 626 ms 20740 KB Output is correct

Subtask #4
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 578 ms 20200 KB Output is correct
2 Correct 600 ms 20240 KB Output is correct
3 Correct 611 ms 21240 KB Output is correct
4 Correct 128 ms 19588 KB Output is correct

Subtask #5
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 267 ms 19600 KB Output is correct
2 Correct 420 ms 21632 KB Output is correct
3 Correct 245 ms 15184 KB Output is correct
4 Correct 579 ms 21124 KB Output is correct

Subtask #6
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 585 ms 18688 KB Output is correct
2 Correct 526 ms 20164 KB Output is correct
3 Correct 518 ms 18968 KB Output is correct
4 Correct 622 ms 22384 KB Output is correct

Subtask #7
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 6065 ms 50668 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #8
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 6046 ms 49356 KB Time limit exceeded
2 Halted 0 ms 0 KB -