/* -----------------------
Autor: Tomasz Boguslawski
-------------------------- */
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<fstream>
#include<iomanip>
#include<string>
#include<sstream>
#include<cstring>
#include<map>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include <fstream>
#include<math.h>
#define LL long long
#define FOR(x, b, e) for(LL x = b; x <= (e); x++)
#define FORS(x, b, e, s) for(LL x = b; x <= (e); x+=s)
#define FORD(x, b, e) for(LL x = b; x >= (e); x--)
#define VAR(v, n) __typeof(n) v = (n)
#define ALL(c) (c).begin(), (c).end()
#define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i)
#define DEBUG if (debug)
#define MIN(a,b) ((a>b)?b:a)
#define MAX(a,b) ((a>b)?a:b)
using namespace std;
LL w[10000];
LL n, k;
/// MAIN
int main(int argc, char* argv[])
{
// magic formula, which makes streams work faster:
ios_base::sync_with_stdio(0);
cin >> n; cin >> k;
FOR(i,1,n) cin >> w[i];
LL sum=0; LL addon=0;
FOR(i,2,n)
{
if (w[i]<w[i-1]-k)
{
addon=w[i-1]-k-w[i]; sum+=addon; w[i]+=addon;
//cout << "add1 to " << i << " amount=" << addon << "\n";
}
else if (w[i]>w[i-1]+k)
{
FORD(j,i-1,1)
{
if (w[j]<w[j+1]-k)
{
addon=w[j+1]-k-w[j]; sum+=addon; w[j]+=addon;
//cout << "add2 to " << j << " amount=" << addon << "\n";
}
}
}
}
cout << sum << "\n";
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | /* ----------------------- Autor: Tomasz Boguslawski -------------------------- */ #include<cstdio> #include<cstdlib> #include<iostream> #include<fstream> #include<iomanip> #include<string> #include<sstream> #include<cstring> #include<map> #include<vector> #include<set> #include<queue> #include<algorithm> #include <fstream> #include<math.h> #define LL long long #define FOR(x, b, e) for(LL x = b; x <= (e); x++) #define FORS(x, b, e, s) for(LL x = b; x <= (e); x+=s) #define FORD(x, b, e) for(LL x = b; x >= (e); x--) #define VAR(v, n) __typeof(n) v = (n) #define ALL(c) (c).begin(), (c).end() #define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i) #define DEBUG if (debug) #define MIN(a,b) ((a>b)?b:a) #define MAX(a,b) ((a>b)?a:b) using namespace std; LL w[10000]; LL n, k; /// MAIN int main(int argc, char* argv[]) { // magic formula, which makes streams work faster: ios_base::sync_with_stdio(0); cin >> n; cin >> k; FOR(i,1,n) cin >> w[i]; LL sum=0; LL addon=0; FOR(i,2,n) { if (w[i]<w[i-1]-k) { addon=w[i-1]-k-w[i]; sum+=addon; w[i]+=addon; //cout << "add1 to " << i << " amount=" << addon << "\n"; } else if (w[i]>w[i-1]+k) { FORD(j,i-1,1) { if (w[j]<w[j+1]-k) { addon=w[j+1]-k-w[j]; sum+=addon; w[j]+=addon; //cout << "add2 to " << j << " amount=" << addon << "\n"; } } } } cout << sum << "\n"; return 0; } |
English