C语言实现 开灯问题

开灯问题。
有n盏灯，编号为1～n。第1个人把所有灯打开，第2个人按下所有编号为2的倍数的开关（这些灯将被关掉），第3个人按下所有编号为3的倍数的开关（其中关掉的灯将被打开，开着的灯将被关闭），依此类推。
一共有k个人，问最后有哪些灯开着？
输入n和k，输出开着的灯的编号。k≤n≤1000。
样例输入：7 3
样例输出：1 5 6 7

---

主要思路

建立一个足够长的数组，由于第一个人把灯全部打开，可以将长度为n的部分赋值为1，代表灯开
而后在循环中，寻找2，4，6…3，6，9…等等位置的开关取反
在循环结束后，遍历数组，输出为1的位置即可

#include<stdio.h>
#include<string.h>
#define maxn 1005
int a[maxn];
int main(){
    int n,k,first=1;
    scanf("%d%d",&n,&k);
    /*for(int i=1;i<=n;i++){
        a[i]=1;
    }*/
    memset(a,1,sizeof(a));//memset(*array,i,siezeof())用于将内存中连续一段赋值
    for(int i=2;i<=k;i++){
        for(int j=i;j<=n;j*=i){
            if(a[j]==0){ 
                a[j]=1;
            }
            else{
                a[j]=0;
            }
        }
    } 
    for(int i=1;i<=n;i++){
        if(a[i]){
            //first变量的作用在于标识第一个输出的i，因为输出格式要求空格间隔，故除了第一个输出外，所有输出前应输出一个空格
            if(first) first=0;
            else      printf(" ");
            printf("%d",i);
        }
    }
    return 0;
}