Submission #124186

#	Time	Username	Problem	Language	Result	Execution time	Memory
124186		mechfrog88	Safety (NOI18_safety)	C++14	100 / 100	284 ms	21128 KiB

safety

#include <iostream>
#include <algorithm>
#include <set>
#include <functional>
using namespace std;
multiset<long long,greater<long long>> lefts;
multiset<long long,less<long long>> rights;
int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    int n,h;
    cin>>n>>h;
    {
        int x;
        cin>>x;
        lefts.insert(x);
        rights.insert(x);
    }
    long long minval=0;
    for(int i=1;i<n;++i){
        int x;
        cin>>x;
        long long shift=static_cast<long long>(i)*h;
        auto leftborder=lefts.begin();
        long long leftborderval=*leftborder-shift;
        auto rightborder=rights.begin();
        long long rightborderval=*rightborder+shift;
        if(x<leftborderval){
            auto hint=lefts.insert(x+shift);
            lefts.insert(hint,x+shift);
            lefts.erase(leftborder);
            rights.insert(rightborder,leftborderval-shift);
            minval+=abs(leftborderval-x);
        }
        else if(x>rightborderval){
            auto hint=rights.insert(x-shift);
            rights.insert(hint,x-shift);
            rights.erase(rightborder);
            lefts.insert(leftborder,rightborderval+shift);
            minval+=abs(rightborderval-x);
        }
        else{
            lefts.insert(leftborder,x+shift);
            rights.insert(rightborder,x-shift);
        }
    }
    cout<<minval<<endl;
}

• Subtask 1: 3 / 3
• Subtask 2: 4 / 4
• Subtask 3: 9 / 9
• Subtask 4: 5 / 5
• Subtask 5: 6 / 6
• Subtask 6: 11 / 11
• Subtask 7: 11 / 11
• Subtask 8: 22 / 22
• Subtask 9: 29 / 29