Submission #1054199

#	Time	Username	Problem	Language	Result	Execution time	Memory
1054199		Kipras	Safety (NOI18_safety)	C++17	7 / 100	28 ms	4568 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;

int main() {

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

    cin>>n>>h;
    ll aa;
    cin>>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 = -5; x < 5; 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: 0 / 5
• Subtask 5: 0 / 6
• Subtask 6: 0 / 11
• Subtask 7: 0 / 11
• Subtask 8: 0 / 22
• Subtask 9: 0 / 29