C++で最大の幾何平均を持つサブセットを検索します

ここに、いくつかの要素を含む配列Aがあります。私たちのタスクは、幾何平均が最大になるサブセットを見つけることです。 A =[1、5、7、2、0]とすると、幾何平均が最大のサブセットは[5、7]になります。

これを解決するために、1つのトリックに従います。最大の2つの要素が最大の幾何平均を形成し、最大の2つの要素がサブセットとして返されることがわかっているため、平均は見つかりません。

例

#include <iostream>
using namespace std;
void largestGeoMeanSubset(int arr[], int n) {
   if (n < 2) {
      cout << "Very few number of elements";
      return;
   }
   int max = INT_MIN, second_max = INT_MIN;
   for (int i = 0; i < n ; i ++) {
      if (arr[i] > max) {
         second_max = max;
         max = arr[i];
      }else if (arr[i] > second_max)
         second_max = arr[i];
   }
   cout << second_max << ", "<< max;
}
int main() {
   int arr[] = {1, 5, 7, 2, 0};
   int n = sizeof(arr)/sizeof(arr[0]);
   largestGeoMeanSubset(arr, n);
}

出力

5, 7