C++에서 arr[i]*i의 합 최대화

문제 설명

N개의 정수 배열이 주어집니다. 배열의 요소를 재정렬할 수 있습니다. 작업은 Σarr[i]*i의 최대값을 찾는 것입니다. 여기서 i =0, 1, 2, .. n – 1입니다.

입력 배열 ={4, 1, 6, 2}인 경우 정렬된 순서로 요소를 재배열하면 최대 합은 28이 됩니다.

{1, 2, 4, 6} = (1 * 0) + (2 * 1) + (4 * 2) + (6 * 3) = 28

알고리즘

1. Sort array in ascending order
2. Iterate over array and multiply each array element by 1 where i = 0, 1, 2, n – 1.
3. Return sum

예

#include <bits/stdc++.h>
using namespace std;
int getMaxSum(int *arr, int n){
   sort(arr, arr + n);
   int sum = 0;
   for (int i = 0; i < n; ++i) {
      sum = sum + arr[i] * i;
   }
   return sum;
}
int main(){
   int arr[] = {4, 1, 6, 2};
   int n = sizeof(arr) / sizeof(arr[0]);
   cout << "Maximum sum = " << getMaxSum(arr, n) << endl;
   return 0;
}

출력

위의 프로그램을 컴파일하고 실행할 때. 다음 출력을 생성합니다-

Maximum sum = 28