Submission #951613

#	Time	Username	Problem	Language	Result	Execution time	Memory
951613		Irate	Pilot (NOI19_pilot)	C++17	100 / 100	871 ms	99764 KiB

pilot

#include<bits/stdc++.h>
using namespace std;
struct Node{
    long long num_of_intervals = 0;
    int pref = 0, suff = 0;
};
struct SegmentTree{
    vector<Node>sTree;
    SegmentTree(int n){
        sTree.resize(4 * n);
    }
    Node Merge(Node &left, Node &right, int &l, int &r){
        Node res;
        int mid = (l + r) >> 1;
        res.pref = left.pref;
        if(left.pref == mid - l + 1) res.pref += right.pref;
        res.suff = right.suff;
        if(right.suff == r - mid) res.suff += left.suff;
        res.num_of_intervals = left.num_of_intervals + right.num_of_intervals;
        res.num_of_intervals -= (long long)right.pref * (right.pref + 1) / 2 + (long long)left.suff * (left.suff + 1) / 2;
        res.num_of_intervals += ((long long)right.pref + left.suff) * (right.pref + left.suff + 1) / 2;
        return res;
    }
    void Update(int node, int l, int r, int pos){
        if(l == r){
            sTree[node] = {1, 1, 1};
        }
        else{
            int mid = (l + r) >> 1;
            if(pos <= mid){
                Update(node * 2, l, mid, pos);
            }
            else{
                Update(node * 2 + 1, mid + 1, r, pos);
            }
            sTree[node] = Merge(sTree[node * 2], sTree[node * 2 + 1], l, r);
        }
    }
};
int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n, q;
    cin >> n >> q;
    vector<pair<int, int>>v(n + 1);
    for(int i = 1;i <= n;++i){
        cin >> v[i].first;
        v[i].second = i;
    }    
    vector<pair<int, int>>queries(q);
    vector<long long> answers(q);
    for(int i = 0;i < q;++i) {
        cin >> queries[i].first;
        queries[i].second = i;
    }
    sort(v.begin(), v.end());
    sort(queries.begin(), queries.end());
    SegmentTree tree(n);
    int p_v = 1, p_queries = 0;
    while(p_queries != q){
        if(p_v <= n && v[p_v].first <= queries[p_queries].first){
            tree.Update(1, 1, n, v[p_v].second);
            p_v++;
        }
        else{
            answers[queries[p_queries].second] = tree.sTree[1].num_of_intervals;
            p_queries++;
        }
    }
    for(int i = 0;i < q;++i){
        cout << answers[i] << '\n';
    }
}

• Subtask 1: 3 / 3
• Subtask 2: 10 / 10
• Subtask 3: 12 / 12
• Subtask 4: 15 / 15
• Subtask 5: 5 / 5
• Subtask 6: 9 / 9
• Subtask 7: 14 / 14
• Subtask 8: 10 / 10
• Subtask 9: 11 / 11
• Subtask 10: 11 / 11