开灯问题

描述

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

输入

输入一组数据：n和k

输出

输出开着的灯编号

样例输入

7 3

样例输出

1 5 6 7

#include<stdio.h>

int main(){
    int k,n,i;
    scanf("%d%d",&n,&k);//k为人数，n为灯
    int a[1000]={0};
    
    
    for (int j=1;j<=k;j++){
        for(i=j;i<=n;i=i+j){
            a[i]=!a[i];
        }
    }
        for (i=1; i<=n;i++ ) {
            if(a[i]==1){
                printf("%d ",i);
   
        }
        
    }
printf("\n");
    return 0;
}