#include <bits/stdc++.h>
using namespace std;
using ll = int64_t;
constexpr ll mod = 1000000007;
void solve() {
int n, k;
cin >> n >> k;
vector<int> vs(n);
for(auto& v : vs) {
cin >> v;
}
vector<int> m = vs;
for(int i = 0; i < n; i++) {
for(int j = 0; j < n; j++) {
int dist = abs(j - i);
int need = vs[i] - dist * k;
m[j] = max(m[j], need);
}
}
int ans = 0;
for(int i = 0; i < n; i++) {
ans += m[i] - vs[i];
}
cout << ans;
}
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
//freopen("in", "r", stdin);
solve();
}
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 | #include <bits/stdc++.h> using namespace std; using ll = int64_t; constexpr ll mod = 1000000007; void solve() { int n, k; cin >> n >> k; vector<int> vs(n); for(auto& v : vs) { cin >> v; } vector<int> m = vs; for(int i = 0; i < n; i++) { for(int j = 0; j < n; j++) { int dist = abs(j - i); int need = vs[i] - dist * k; m[j] = max(m[j], need); } } int ans = 0; for(int i = 0; i < n; i++) { ans += m[i] - vs[i]; } cout << ans; } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); //freopen("in", "r", stdin); solve(); } |
English