Минимальное количество действий, чтобы сделать все башенки одинаковой высоты

#include <iostream>
using namespace std;
int main() {
  int n, a[1000];
  cin >> n;
  for (int i = 0; i < n; i++) {
    cin >> a[i];
  }
  int max = 0, min = 0;
  for (int i = 1; i < n; i++) {
    if (a[i] > a[max]) max = i;
    if (a[i] < a[min]) min = i;
  }
  int height = (a[max] + a[min]) / 2;
  int count = 0;
  for (int i = 0; i < n; i++) {
    if (i != max) {
      if (a[i] < height) {
        while (a[i] != height) {
          a[i]++;
          a[max]--;
          count++;
        }
      } else if (a[i] > height) {
        while (a[i] != height) {
          a[i]--;
          a[min]++;
          count++;
        }
      }
      min = 0;
      max = 0;
      for (int j = 1; j < n; j++) {
        if (a[j] > a[max]) max = j;
        if (a[j] < a[min]) min = j;
      }
    }
  }
  for (int i = 0; i < n; i++) {
    if (a[i] < height) {
      while (a[i] != height) {
        a[i]++;
        count++;
      }
    } else if (a[i] > height) {
      while (a[i] != height) {
        a[i]--;
        count++;
      }
    }
  }
  cout<<count;
  return 0;
}

int ScroogeMyDuck (int n, int *a)
{
  // переменные
  int i, j, sum = 0, avg, count = 0, cube_set = 0;
  double tmp, dif;
 
  // считаем кол-во всех кубиков в башнях
  for (i = 0; i < n; i++)
    sum += a[i];
 
  tmp = 1.0 * sum / n;  // среднее значение
  dif = tmp - int(tmp);  // разница с меньшим целым
 
  // среднее целое значение - "округляем" точное значение до ближайшего целого (как в раньше в школе делали)
  avg = int(tmp) + (dif < 0.5 ? 0 : 1);
 
  // сортируем значения (по убыванию или возрастанию) - это может влиять на лимит по времени
  for (i = 0; i < n - 1; i++)
    for (j = i + 1; j < n; j++)
      if (dif < 0.5 && a[j] > a[i] || dif >= 0.5 && a[j] < a[i])
      {
        int tmp2 = a[j];
        a[j] = a[i];
        a[i] = tmp2;
      }
 
  // "проходим по башням"
  for (i = 0; i < n; i++)
  {
    j = a[i] - avg;  // разница в кубиках между i-ой башней и средним значением
 
    cube_set += j;  // кол-во кубиков, доступных для перемещения (КДДП)
 
    // если есть кубики (с башен), которые можно переместить
    if (dif < 0.5 && cube_set > 0 && j > 0 || dif >= 0.5 && cube_set < 0 && j < 0)
    {
      count += abs(j);  // кол-во действий
      cube_set -= j;  // уменьшаем кол-во КДДП
    }
  }
 
  return count;
}