oj.uz

Submission #1078045

# Submission time Handle Problem Language Result Execution time Memory
1078045 2024-08-27T11:54:13 Z 0npata Fish 2 (JOI22_fish2) C++17
0 / 100
 0 ms 348 KB 
#include<bits/stdc++.h>
using namespace std;


#define vec vector
#define int long long

const int INF = 1e17;

int32_t main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);

    int N;
    cin >> N;


    vec<int> A(N);
    set<pair<int, int>> A_s;

    for(int i = 0; i<N; i++) {
        cin >> A[i];
        A_s.insert({-A[i], i});
    }

    int Q;
    cin >> Q;
    while(Q--) {
        //cerr << "f" << '\n';
        int X, Y;
        cin >> X >> Y;
        X--;
        A_s.erase({-A[X], X});
        A[X] = Y;
        A_s.insert({-Y, X});

        vec<int> l(N);
        vec<int> r(N);

        vec<pair<int, int>> hi{{INF, -1}};

        for(int i = 0; i<N; i++) {
            while(hi.back().first <= A[i]) {
                hi.pop_back();
                //cerr << "here1" << '\n';
            }
            l[i] = hi.back().second;
            hi.push_back({A[i], i});
        }

        vec<pair<int, int>> hi_r{{INF, N}};

        for(int i = N-1; i>=0; i--) {
            while(hi_r.back().first < A[i]) {
                //cerr << "here2" << '\n';
                hi_r.pop_back();
            }
            r[i] = hi_r.back().second;
            hi_r.push_back({A[i], i});
        }

        vec<bool> valid(N);
        vec<int> prefixsums(N+2);
        for(int i = 1; i<=N; i++) {
            prefixsums[i] = prefixsums[i-1] + A[i-1];
        }
        prefixsums[N+1] = prefixsums[N];

        valid[(*A_s.rbegin()).second] = true;
        int ans = 1;
        for(auto [_, i] : A_s) {
            int cl = l[i];
            int cr = r[i];

            int sum = prefixsums[cr+1]-prefixsums[cl+1];

            if(cl != -1) {
                if(sum >= A[cl]) valid[i] = valid[i] | valid[cl];
            }
            if(cr != N) {
                if(sum >= A[cr]) valid[i] = valid[i] | valid[cr];
            }
            ans += valid[i];
        }

        cout << ans << '\n';
    }

}

Subtask #1
0 / 5.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 8.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 348 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 12.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 23.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 348 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
0 / 35.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 348 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
0 / 17.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB Output isn't correct
2 Halted 0 ms 0 KB -