Задан целочисленный массив, содержащий N элементов. Найти номера индекса, соответствующего среднему арифметическому знач

#include <iostream>
 
using namespace std;
 
int main()
{
    int N;
    cout << "Input array length : ";
    cin >> N;
    int *arr = (int*)malloc(N * sizeof(int));
    for (int index = 0; index < N; index++){
        cout << "A[" << index << "]=";
        cin >> arr[index];
    }
    int sumAboveZero = 0;
    int countElemtsAboveZero = 0;
    for (int index = 0; index < N; index++){
        if (arr[index] > 0){
            sumAboveZero += arr[index];
            countElemtsAboveZero++;
        }
    }
    if (countElemtsAboveZero){ //если есть положительные элементы
        double average = (double)sumAboveZero / countElemtsAboveZero;
        cout << "Average of positive values = " << average << endl;
        if (floor(average) != average) //если ср. арифм. - дробное число, то нет смысыла проверять на равенство
            cout << "Average is real " << endl; 
        else{ 
            for (int index = 0; index < N; index++)
                if (arr[index] == average)
                    cout << index << "  ";
        }
        cout << endl;
    }
    else //если все элементы отрицательные
        cout << "Elements above zero not found " << endl;
    free(arr);
    system("pause");
    return 0;
}