Submission #960778

#	Time	Username	Problem	Language	Result	Execution time	Memory
960778		Zero_OP	Lightning Conductor (POI11_pio)	C++17	72 / 100	1047 ms	44368 KiB

pio

#include<bits/stdc++.h>

using namespace std;

int ceil_sqrt(int n){
    int x = sqrt(n);
    while(x * x < n) ++x;
    return x;
}

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

    int n; cin >> n;
    vector<int> h(n);
    vector<vector<int>> st;
    st.push_back(vector<int>(n));
    for(int i = 0; i < n; ++i){
        cin >> h[i];
        st[0][i] = h[i];
    }

    for(int i = 1; (1 << i) <= n; ++i){
        st.push_back(vector<int>(n - (1 << i) + 1));
        for(int j = 0; j + (1 << i) <= n; ++j){
            st[i][j] = max(st[i - 1][j], st[i - 1][j + (1 << (i - 1))]);
        }
    }

    auto max_range = [&](int l, int r){
        int b = 31 - __builtin_clz(r - l  + 1);
        return max(st[b][l], st[b][r - (1 << b) + 1]);
    };

    vector<int> p(n);
    for(int i = 0; i < n; ++i){ 
        if(i != 0){
            int B = ceil_sqrt(i);
            for(int k = 1; k <= B; ++k){
                p[i] = max(p[i], max_range(max(0, i - k * k), i - (k - 1) * (k - 1) - 1) + k - h[i]);
            }
        }

        if(i != n - 1){
            int B = ceil_sqrt(n - 1 - i);
            for(int k = 1; k <= B; ++k){
                p[i] = max(p[i], max_range(i + (k - 1) * (k - 1) + 1, min(n - 1, i + k * k)) + k - h[i]);
            }
        }
    }

    for(int i = 0; i < n; ++i){
        cout << p[i] << '\n';
    }
    
    return 0;
}

• Subtask 1: 9 / 9
• Subtask 2: 9 / 9
• Subtask 3: 9 / 9
• Subtask 4: 9 / 9
• Subtask 5: 9 / 9
• Subtask 6: 9 / 9
• Subtask 7: 9 / 9
• Subtask 8: 9 / 9
• Subtask 9: 0 / 9
• Subtask 10: 0 / 9
• Subtask 11: 0 / 10