开灯问题

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

 

代码

# include <stdio.h>
# include <string.h>
# define max 1005
int a[max];

int main(){
    int n,k;
    memset(a,0,sizeof(a));
    scanf("%d%d",&n,&k);
    for(int i = 2;i <= k; i++){
        for(int j = 2;j <= n; j++){
            if(j%i==0){
                a[j] = !a[j];
            }
        }
    }
    for(int i = 1;i <= n; i++){
        if(!a[i])
            printf("%d\t",i);
    }
}