#include <iostream>
#include <vector>
using namespace std;
void czytaj(vector<int> *A, vector<int> *indeksyMalejace,int n, int k) {
for(int i = 0; i < n; i++) {
int pom;
scanf("%d", &pom);
A->push_back(pom);
indeksyMalejace->push_back(i);
}
}
void sortuj(vector<int> A, vector<int> *indeksyMalejace, int n) {
for(int i = 0; i < n; i++) {
for(int j = 0; j < n; j++) {
if(A[j]<A[i]) {
swap(A[i], A[j]);
swap((*indeksyMalejace)[i], (*indeksyMalejace)[j]);
}
}
}
}
int wyrownaj(int i, int dol, int j, int gora, int k) {
return max(0, (gora - dol - abs(i - j) * k) );
}
int ileDosypac(vector<int> A, vector<int> indeksyMalejace, int n, int k) {
int sumaZwiru = 0;
for(int i = 0; i < n; i++) {
int min = 0;
for(int j = 0; j < i; j++) {
min = max(min, wyrownaj(indeksyMalejace[i], A[indeksyMalejace[i]], indeksyMalejace[j], A[indeksyMalejace[j]], k));
}
sumaZwiru += min;
A[indeksyMalejace[i]] += min;
}
return sumaZwiru;
}
int main() {
int n, k;
scanf("%d%d", &n, &k);
vector<int> A, indeksyMalejace;
czytaj(&A, &indeksyMalejace, n, k);
sortuj(A, &indeksyMalejace, n);
printf("%d\n", ileDosypac(A, indeksyMalejace, n, k));
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 | #include <iostream> #include <vector> using namespace std; void czytaj(vector<int> *A, vector<int> *indeksyMalejace,int n, int k) { for(int i = 0; i < n; i++) { int pom; scanf("%d", &pom); A->push_back(pom); indeksyMalejace->push_back(i); } } void sortuj(vector<int> A, vector<int> *indeksyMalejace, int n) { for(int i = 0; i < n; i++) { for(int j = 0; j < n; j++) { if(A[j]<A[i]) { swap(A[i], A[j]); swap((*indeksyMalejace)[i], (*indeksyMalejace)[j]); } } } } int wyrownaj(int i, int dol, int j, int gora, int k) { return max(0, (gora - dol - abs(i - j) * k) ); } int ileDosypac(vector<int> A, vector<int> indeksyMalejace, int n, int k) { int sumaZwiru = 0; for(int i = 0; i < n; i++) { int min = 0; for(int j = 0; j < i; j++) { min = max(min, wyrownaj(indeksyMalejace[i], A[indeksyMalejace[i]], indeksyMalejace[j], A[indeksyMalejace[j]], k)); } sumaZwiru += min; A[indeksyMalejace[i]] += min; } return sumaZwiru; } int main() { int n, k; scanf("%d%d", &n, &k); vector<int> A, indeksyMalejace; czytaj(&A, &indeksyMalejace, n, k); sortuj(A, &indeksyMalejace, n); printf("%d\n", ileDosypac(A, indeksyMalejace, n, k)); return 0; } |
English