Ввести ряд чисел и сформировать списки элементов ряда, меньших или больших среднего геометрического элементов

int main()
{
    int N;
    int A[10];
    int i;
    double  res = 1;
    int min = 0, max = 0;
 
    cout << "Vvedite razmer:";
    cin >> N;
    for (i = 0; i < N; ++i)
        cin >> A[i];
    for (i = 0; i < N; ++i)
        res *= A[i];
    res = pow(res, 1.0 / N);
    cout << "\n res = " << res<<"   ";
 
    return 0;
}

#include <iostream>
#include <cmath>
#include <vector>
 
using namespace std;
 
int main()
{
    int N;
    int A[10];
    int i;
    double res = 1;
    int min = 0, max = 0;
    vector<int> less, greater; // создаем векторы для элементов меньших и больших среднего геометрического
 
    cout << "Vvedite razmer:";
    cin >> N;
    for (i = 0; i < N; ++i)
        cin >> A[i];
    for (i = 0; i < N; ++i)
        res *= A[i];
    res = pow(res, 1.0 / N);
    cout << "\n res = " << res << endl;
 
    for (i = 0; i < N; ++i) { // цикл для поиска элементов меньших среднего геометрического
        if (A[i] < res)
            less.push_back(A[i]);
    }
    cout << "Elements less than geometric mean: ";
    for (int j = 0; j < less.size(); j++)
        cout << less[j] << " ";
    cout << endl;
 
    for (i = 0; i < N; ++i) { // цикл для поиска элементов больших среднего геометрического
        if (A[i] > res)
            greater.push_back(A[i]);
    }
    cout << "Elements greater than geometric mean: ";
    for (int j = 0; j < greater.size(); j++)
        cout << greater[j] << " ";
    cout << endl;
 
    return 0;
}