Определить количество ненулевых элементов массива начиная с последнего и заканчивая Ai

#include<iostream>
#include<cmath>
 
void func(int* array, int size)
{
    int *sqrts = new int[size]{0}; // Массив с корнями
    int *nzelem = new int[size]{0}; // Массив с числом ненулевых элементов от конца массива до текущего элемента
    for(int i = size - 1; i >= 0; --i) // Считаем
    {
        if(array[i] <= 0)
        {
            sqrts[i] = 0;
            if(i != size - 1)
                nzelem[i] = nzelem[i + 1];
        }
        else 
        {
            sqrts[i] = sqrt(array[i]);
            if(i != size - 1)
                nzelem[i] = nzelem[i + 1];
            nzelem[i]++;
        }
    }
    
    for (int i = 0; i < size; ++i) // Выводим
        std::cout << sqrts[i] << ' ' << nzelem[i] << '\n';
        
    delete[] sqrts;
    delete[] nzelem;
}
 
int main()
{
    int array[3] = {2, 64, 9};
    int* array2 = new int[5] {-6, -78, 0, 0, 1};
    
    func(array, 3);
    
    std::cout << "\n\n";
    
    func(array2, 5);
    
    return 0;
}

#include <iostream>
#include <cmath>
#include <cstdlib>
#include <ctime>
using namespace std;
 
int main()
{
    setlocale(LC_ALL, "ru");
    srand((int)time(0));
 
    int N = 0, i, j;
    cout << "Вводите количество целых чисел (должна больше 0):" << endl;
    while (N < 1)
    {
        cout << "N = ";
        cin >> N;
    }
 
    cout << "Целые числа случайнами -1000...1000:" << endl;
    int* A = new int[N];
    for (i = 0; i < N; i++)
    {
        A[i] = rand() % 2001 - 1000;
        cout << A[i] << "  ";
    }
    cout << endl;
 
    cout << "Суммы Ai:" << endl;
    long double sumA2 = 0.0;
    for (i = 0; i < N; i++)
    {
        if (A[i] < 0)
            sumA2 = 0;
        else
        {
            int k = 0;
            for (j = N - 1; j >= i; j--)
            if (A[j])
                k++;
            sumA2 = sqrt(A[i]) + k;
        }
        cout << sumA2 << "  ";
    }
 
    delete[] A;
 
    cout << endl;
    system("pause");
    return 0;
}