Сформировать массив, поместив в него значения индексов тех элементов первого массива, которые являются простыми числами

[Kapralick]

Доброго времени суток. Мне дали задание:"Дан целочисленный массив A[N]. Сформировать массив B[M], поместив в него значения индексов тех элементов массива A, которые являются простыми числами." Я написал под это задание программу:

#include "stdafx.h" 
#include<math.h> 
#include <conio.h> 
#include <iostream> 
#include <cstdlib> 
using namespace std;
int main()
{
    setlocale(LC_ALL, "Russian");
    int A[100];
    int B[100];
    int i, j, N, M = 0;
    cout << "введите размер массива N:";
    cin >> N;
    for (i = 0; i < N; i++)
    {
        cout << "введите элемент массива A[" << i << "]:";
        cin >> A[i];
    }
    for (i = 0; i < N; i++)
    {
        for (j = 2; j < A[i]; j++)
        {
            if (A[i] % j == 0)
            {
                A[i] = 1;
            }
        }
 
    }
    for (i = 0; i < N; i++)
    {
        if (A[i] != 1)
        {
            B[M] = i;
            cout << "B[" << M << "]=" << B[M] << endl;
            M++;
        }
    }
    system("pause");
}

Возник вопрос:как можно переделать эту программу, она работает правильно, но сам вид не удовлетворяет преподавателя.

[Croessmah]

Так у него и спросите, что ему не нравится.

[p51x]

Вынести проверку простоты в функцию
Либо динамические массивы, либо проверять, что N < 100

[Serge_Bliznykov]

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

ну и вынести проверку на простоту в отдельную функцию, имхо, хорошая идея:

    int m=-1;
    for (i = 0; i < N; i++)
    {
        if(isSimple(A[i])
            B[++m]=i;
    }

плюс, в конце, перед выводом массива B, добавить проверку на то, что простых чисел в исходном массиве вообще не было (if (m<0) cout << "Нет простых чисел в исходном массиве";