Nhập vào số nguyên N rồi xuất ra màn hình N số nguyên tố

Nhập vào số nguyên N rồi xuất ra màn hình N số nguyên tố

#include
#include
#include
#include
 
int ktSNT (const int a)
{
    int i = 0;
    for(i = 2; i <= (int)sqrt(a); i++)
    {
        if( a % i == 0 || a < 2 && a != 2)
        {
            return 0;
        }
    }
    return 1;
}
int main ()
{
    int N,i, k = 0;
    int* snt = 0;
    printf("\n Nhap vao N:");
    scanf("%d", &N);
 
    snt = (int*)malloc(N*sizeof(int));
    k = 0;
    i = 2;
    while(k < N)
    {
        i++;
        if(ktSNT(i) == 1)
        {
            snt[k] = i;
            k++;
        }
    }
 
    printf("\n          Bang cac so nguyen to\n");
    for (i = 0; i < k; i++)
    {
        printf("%5d", snt[i]);
    }
    free(snt);
    getch();
    return 0;
}