开灯问题

问题

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

解决

#include<string.h>
#define MAX 1010
int lights[MAX];
int main(){
    int n,k;
    scanf("%d %d",&n,&k);
    //all lights off
    memset(lights,0,sizeof(lights));
    for(int i=1;i<=k;++i){
        for(int j=1;j<=n;++j){
            if(j%i==0) lights[j]=!lights[j];
        }
    }
    for(int i=1;i<=n;++i){
        if(lights[i]) printf("%d ",i);
    }
    printf("\n");
    return 0;
}```