开灯问题（C语言）

开灯问题，有n盏灯，编号为1~n。 第一个人把所有灯都打开，第二个人按下所有编号为2的倍数的开关（这些灯将被关掉），第三个人按下所有编号为3的倍数的开关（其中关掉的灯将被打开，开着的灯将被关闭），以此类推，一共有k个人

代码如下：

#include <stdio.h>
#include <stdlib.h>
#include<string.h>
int main()
{
    int a[1000],n,k;
    memset(a,0,sizeof(a));//将集合a里的元素全部赋值为0，初始状态为关灯
    scanf("%d%d",&n,&k);
    for(int i=1;i<=k;i++){
        for(int j=1;j<=n;j++){
            if(j%i==0){//如果j%i==0，关的灯打开，开的灯关上，例如当i=1时，关的灯都打开，i=2时，2倍数的开关都关上（因为i=1时灯全部为打开状态），依次类推
                a[j]=!a[j];
            }
        }
    }
    for(int i=1;i<=n;i++){
        if(a[i]){//如果灯开着，则输出灯的编号
            printf("%d ",i);
        }
    }
    return 0;
}