开灯问题

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

样例输入

7 3

样例输出

1 5 6 7

代码如下

#include<stdio.h>
#include<stdlib.h>
int main() {
	int n,m,i,k;
	scanf("%d%d",&n,&k);
	int a[n+1]={0};
	for(int j=1;j<=k;j++){
		i=j;
		while(i<=n){
			if(a[i]==0){
				a[i]=1;
			}
			else{
				a[i]=0;
			}
			i=i+j;
		}
	}
	
	for(i=1;i<=n;i++){
		if(a[i]==1){
			printf("%d\t",i);
		}
	}
	
	return 0;
}