Submission #1054240

#	Time	Username	Problem	Language	Result	Execution time	Memory
1054240		Kipras	Safety (NOI18_safety)	C++17	12 / 100	2093 ms	2632 KiB

safety

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll, ll> pl;

const ll maxC = 5010;

ll n, h;
vector<ll> diffs;

ll res = 1e18;
ll median = 0;
void solve() {
    vector<ll> a;
    for(int i = 0; i < n; i++) {
        ll aa;
        cin>>aa;
        a.push_back(aa);
    }
    sort(a.begin(), a.end());
    median=a[n/2];
    ll lRes=0;
    for(auto i : a) {
        lRes+=abs(i-median);
    }
    if(n%2==0) {
        median=a[(n+2)/2];
        res=0;
        for(auto i : a) {
            res+=abs(i-median);
        }
    }

    cout<<min(lRes, res)<<"\n";
}

int main() {

    ios_base::sync_with_stdio(false);cin.tie(nullptr);

    cin>>n>>h;
    if(h==0) {
        solve();
        return 0;
    }
    ll aa;
    ll a0;
    cin>>aa;
    a0=aa;
    for(int i = 0; i < n-1; i++) {
        ll bb;
        cin>>bb;
        diffs.push_back(bb-aa);
        //cout<<diffs.back()<<" ";
        aa=bb;
    }
    //cout<<"\n\n\n";

    for(int x = -maxC; x <= a0; x++) {
        //cout<<"doing: "<<x<<"\n";
        vector<ll> tmp;
        for(int i = 0; i < n-1; i++) {
            tmp.push_back(diffs[i]);
        }

        ll lRes = abs(x);
        tmp[0]+=x;
        tmp.push_back(0);

        for(int i = 0; i < n-1; i++) {
            if(tmp[i]==0)continue;
            ll v = abs(tmp[i]);
            if(h>=v)continue;
            ll di = v-h;
            if(tmp[i]<0) {
                tmp[i]+=di;
                tmp[i+1]-=di;
            }else if(tmp[i]>0) {
                tmp[i]-=di;
                tmp[i+1]+=di;
            }
            //cout<<i<<" "<<di<<" "<<tmp[i]<<"\n";
            lRes+=di;
        }
        res=min(res, lRes);
    }

    cout<<res<<endl;
    return 0;

}

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