Submission #1054226

#	Time	Username	Problem	Language	Result	Execution time	Memory
1054226		Kipras	Safety (NOI18_safety)	C++17	0 / 100	8 ms	2520 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);
        median+=aa;
    }
    median=(median+n/2)/n;
    res = 0;
    for(auto i : a) {
        res+=abs(i-median);
    }
    cout<<res<<"\n";
}

int main() {

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

    cin>>n>>h;
    if(h==0) {
        solve();
        return 0;
    }
    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";


    ll lowB=0, highB=0;

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


    cout<<res<<endl;
    return 0;

}

Compilation message (stderr)

safety.cpp: In function 'int main()':
safety.cpp:50:8: warning: variable 'lowB' set but not used [-Wunused-but-set-variable]
   50 |     ll lowB=0, highB=0;
      |        ^~~~
safety.cpp:50:16: warning: variable 'highB' set but not used [-Wunused-but-set-variable]
   50 |     ll lowB=0, highB=0;
      |                ^~~~~

• Subtask 1: 0 / 3
• Subtask 2: 0 / 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