Все строки матрицы, где находится более двух простых чисел, обнулить. Найти среднее арифметическое элементов до и после обнуления

#include <iostream>
#include <ctime>
#include <cmath>
#include <iomanip>
using namespace std;
 
bool prime(int num)
{
    if(num < 2)
    {
        return false;
    }
    for(int i = 2; i <= sqrt((float)num); i++)
    {
        if(!(num % i))
        {
            return false;
        }
    }
    return true;
}
 
int main()
{
    srand(time(0));
    cout << "Matrix A[n][n]" << endl;
    int n, sum = 0, quant = 0, rows = 0;
    float before, after;
    cout << "Enter n: "; cin >> n;
    int** A = new int*[n];
    for(int i = 0; i < n; i++)
        A[i] = new int[n];
        
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j < n; j++)
        {
            A[i][j] = rand() % 100 + 1;
        }
    }
    
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j < n; j++)
        {
            cout << setw(5) << A[i][j];
            int temp = A[i][j];
            sum += temp;
        }
        cout << "\n\n";
    }
    
    before = (float)sum / (n * n);
    cout << "\nAverage before: " << before << "\n\n";
    
    for(int i = 0; i < n; i++)
    {
        int counter = 0;
        for(int j = 0; j < n; j++)
        {
            int temp = A[i][j];
            if(prime(temp))
            {
                counter++;
            }
        }
        if(counter >= 2)
        {
            for(int j = 0; j < n; j++)
            {
                A[i][j] = 0;
            }
            rows++;
        }
    }
    
    sum = 0;
    
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j < n; j++)
        {
            cout << setw(5) << A[i][j];
            int temp = A[i][j];
            sum += temp;
            if(temp > 0)
            {
                quant++;
            }
        }
        cout << "\n\n";
    }
    
    after = (float)sum / quant;
    cout << "\nAverage after: " << after << "\n\n";
    cout << "Rows zeroed: " << rows << "\n\n";
    
    system("pause");
    return 0;
}