oj.uz

답안 #964324

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
964324 2024-04-16T16:07:32 Z Pekiban Meteors (POI11_met) C++17
100 / 100
 790 ms 36824 KB 
#include <bits/stdc++.h>
 
using namespace std;
using ll = long long;
const int mxN = 3e5+3;
vector<int> a[mxN];
array<int, 3> e[mxN];
ll bit[mxN], p[mxN], ans[mxN], c = -1;
void upd(int idx, int x) {
    while (idx < mxN) {
        bit[idx] += x;
        idx |= (idx + 1);
    }
}
ll sum(ll idx) {
    ll s = 0;
    while (idx >= 0) {
        s += bit[idx];
        idx &= (idx + 1);
        --idx;
    }
    return s;
}
void supd(int l, int r, int x) {
    upd(l, x);
    upd(r+1, -x);
}
int n, m;
void solve(int l, int r, vector<int> &b) {
    int mid = (l + r)/2;
    while (c < mid) {
        ++c;
        if (e[c][0] <= e[c][1]) {
            supd(e[c][0], e[c][1], e[c][2]);
        }
        else {
            supd(e[c][0], m-1, e[c][2]);
            supd(0, e[c][1], e[c][2]);
        }
    }
    while (c > mid) {
        if (e[c][0] <= e[c][1]) {
            supd(e[c][0], e[c][1], -e[c][2]);
        }
        else {
            supd(e[c][0], m-1, -e[c][2]);
            supd(0, e[c][1], -e[c][2]);
        }
        --c;
    }
    vector<int> L, R;
    for (auto x : b) {
        ll s = 0;
        for (auto i : a[x]) {
            s += sum(i);
            if (s >= p[x])  break;
        }
        if (s >= p[x]) {
            ans[x] = mid;
            L.push_back(x);
        }
        else {
            R.push_back(x);
        }
    }
    b.clear();
    if (l < r) {
        solve(l, mid-1, L);
        solve(mid+1, r, R);
    }
}
int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin >> n >> m;
    fill(ans, ans+n, -1);
    for (int i = 0; i < m; ++i) {
        int x;
        cin >> x;
        --x;
        a[x].push_back(i);
    }
    for (int i = 0; i < n; ++i) {
        cin >> p[i];
    }
    int k;
    cin >> k;
    for (int i = 0; i < k; ++i) {
        cin >> e[i][0] >> e[i][1] >> e[i][2];
        --e[i][0], --e[i][1];
    }
    vector<int> b(n);
    iota(b.begin(), b.end(), 0);
    solve(0, k-1, b);
    for (int i = 0; i < n; ++i) {
        cout << (ans[i] == -1 ? "NIE" : to_string(ans[i] + 1)) << '\n';
    }
}

Subtask #1
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 4 ms 7512 KB Output is correct
2 Correct 5 ms 7772 KB Output is correct
3 Correct 3 ms 7740 KB Output is correct

Subtask #2
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 7516 KB Output is correct
2 Correct 3 ms 7772 KB Output is correct
3 Correct 3 ms 7768 KB Output is correct

Subtask #3
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 36 ms 10232 KB Output is correct
2 Correct 94 ms 12588 KB Output is correct
3 Correct 65 ms 10720 KB Output is correct

Subtask #4
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 72 ms 10320 KB Output is correct
2 Correct 72 ms 10332 KB Output is correct
3 Correct 87 ms 12056 KB Output is correct
4 Correct 20 ms 9808 KB Output is correct

Subtask #5
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 65 ms 10020 KB Output is correct
2 Correct 85 ms 12116 KB Output is correct
3 Correct 63 ms 8796 KB Output is correct
4 Correct 58 ms 10904 KB Output is correct

Subtask #6
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 76 ms 11864 KB Output is correct
2 Correct 85 ms 10500 KB Output is correct
3 Correct 47 ms 10072 KB Output is correct
4 Correct 78 ms 11600 KB Output is correct

Subtask #7
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 547 ms 29932 KB Output is correct
2 Correct 307 ms 20804 KB Output is correct
3 Correct 355 ms 12992 KB Output is correct
4 Correct 671 ms 33904 KB Output is correct

Subtask #8
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 508 ms 28168 KB Output is correct
2 Correct 442 ms 19768 KB Output is correct
3 Correct 290 ms 13392 KB Output is correct
4 Correct 790 ms 36824 KB Output is correct