算法竞赛，第三章，开灯问题

开灯问题
描述

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

输入

输入一组数据：n 和 k

输出

输出开着的灯编号

样例输入

7 3
样例输出

1 5 6 7

#include<iostream>
#include<string.h>
#include<cstdio>
using namespace std;
#define max 1002 
int a[max];  
int main(){
	int k,n;//k为人，n为灯 
	int flag=1;
	memset(a,0,sizeof(a));
	scanf("%d%d",&n,&k);
	for(int i=1;i<=k;i++){
		for(int j=1;j<=n;j++)
		{
			if(j%i==0)a[j]=!a[j];
		}
	}
	for(int i=1;i<=n;i++){
		
		if(a[i]){cout<<i<<" ";}
	}
	
return 0;
	
	
}