Problem_1619_trimMean.cc

#include <algorithm>
#include <iostream>
#include <vector>

#include "UnitTest.h"

using namespace std;

class Solution
{
 public:
  double trimMean(vector<int> arr)
  {
    std::sort(arr.begin(), arr.end());
    int sum = 0;
    int N = arr.size();
    for (int i = N / 20; i < N * 19 / 20; i++)
    {
      sum += arr[i];
    }
    return sum * 1.00000 / (N * 0.9);
  }
};

void testTrimMean()
{
  Solution s;
  vector<int> arr1 = {1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 3};
  vector<int> arr2 = {6, 2, 7, 5, 1, 2, 0, 3, 10, 2, 5, 0, 5, 5, 0, 8, 7, 6, 8, 0};
  vector<int> arr3 = {6, 0, 7, 0, 7, 5, 7, 8, 3, 4, 0, 7, 8, 1, 6, 8, 1, 1, 2, 4, 8, 1, 9, 5, 4, 3, 8, 5, 10, 8, 6, 6, 1, 0, 6, 10, 8, 2, 3, 4};
  vector<int> arr4 = {9, 7, 8,  7, 7,  8, 4, 4,  6, 8, 8, 7, 6, 8, 8, 9, 2, 6, 0, 0, 1, 10, 8, 6, 3, 3, 5, 1, 10, 9,
                      0, 7, 10, 0, 10, 4, 1, 10, 6, 9, 3, 6, 0, 0, 2, 7, 0, 6, 7, 2, 9, 7,  7, 3, 0, 1, 6, 1, 10, 3};
  vector<int> arr5 = {4, 8, 4, 10, 0, 7, 1, 3, 7, 8, 8, 3,  4, 1,  6, 2, 1,  1, 8, 0, 9, 8, 0, 3, 9,  10, 3, 10, 1, 10, 7, 3, 2,  1, 4, 9, 10, 7, 6, 4,
                      0, 8, 5, 1,  2, 1, 6, 2, 5, 0, 7, 10, 9, 10, 3, 7, 10, 5, 8, 5, 7, 6, 7, 6, 10, 9,  5, 10, 5, 5,  7, 2, 10, 7, 7, 8, 2,  0, 1, 1};

  EXPECT_EQ_DOUBLE(2.00000, s.trimMean(arr1));
  EXPECT_EQ_DOUBLE(4.00000, s.trimMean(arr2));
  EXPECT_EQ_DOUBLE(4.77778, s.trimMean(arr3));
  EXPECT_EQ_DOUBLE(5.27778, s.trimMean(arr4));
  EXPECT_EQ_DOUBLE(5.29167, s.trimMean(arr5));
  EXPECT_SUMMARY;
}

int main()
{
  testTrimMean();
  return 0;
}