开灯问题——数组

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

#include<stdio.h>
#include<string.h> 
#define maxn 1001

int main(){
	int a[maxn]={0};
	int n,k,first=1;	//first用来标记是否是第一个， 最后一个输出不能带空格
	scanf("%d%d",&n,&k);
	for(int i=1;i<=k;i++)
		for(int j=i;j<=n;j+=i)
			a[j]=!a[j];
	for(int i=1;i<=n;i++){
		if(a[i]){
			if(first){
				first=0;
			}else{
				printf(" ");
			} 
			printf("%d",i);
		}
	}
	return 0; 
}