编写函数求素数并放在数组中

编写函数fun，函数的功能是求出小于或等于lim的所有素数并放在aa数组中，函数返回所求的素数的个数

下面是一个实现，供参考：

#include <stdio.h>

int primeInLim(int lim){
    
    int i,j,k=0;
    int aa[lim];
    int num=0;
    int primeFlag = 1;
    
    for(i=2;i<=lim;i++){
        
        for(j=2;j<i;j++){
            if(i%j==0){
                primeFlag=0;
            }
        }
        if(primeFlag==1){
            aa[k]=i;
            k++;
        }
        
        primeFlag=1;        
    }
    
    return k;
} 


int main(void){
    
    int test ;
    scanf("%d",&test);
    
    printf("小于等于test的所有素数有%d个。\n",primeInLim(test));
    
    return 0;
}