Submission #623415

#	Time	Username	Problem	Language	Result	Execution time	Memory
623415		PoonYaPat	Safety (NOI18_safety)	C++14	0 / 100	15 ms	2000 KiB

safety

#include <bits/stdc++.h>
using namespace std;

int n,k,a[200001],h[200001],mmax,mm;
long long ans,sum;
map<int,int> cnt;

int main() {
  ios_base::sync_with_stdio(0); cin.tie(0);
  cin>>n>>k;
  for (int i=1; i<=n; ++i) cin>>h[i],sum+=h[i];
  sum/=n;
  if (k==0) {
    for (int i=1; i<=n; ++i) ans+=abs(sum-h[i]);
    cout<<ans;
    return 0;
  }
  a[n]=h[n];
  for (int i=n; i>1; --i) {
    if (a[i]-h[i-1]>k) a[i-1]=a[i]-k;
    else if (h[i-1]-a[i]>k) a[i-1]=a[i]+k;
    else a[i-1]=h[i-1];
    ans+=abs(a[i-1]-h[i-1]);
  }
  cout<<ans;
}

• 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