开灯问题

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

      样例输入：7 3

      样例输出：1 5 6 7

       memset(a,0,sizeof(a));将数组全部赋初值0。  

#include<iostream>
#include<cstring>
#define N 1010
int a[N];
using namespace std;
int main()
{
   	int n,k;           //n灯，k人.
   	int first=1;
	memset(a,0,sizeof(a));    //0表示关，1表示开 ，memset(a,0,sizeof(a));将数组全部赋初值0。  
	cin>>n>>k;
	for(int i=1;i<=k;i++)
	{
	    for(int j=1;j<=n;j++)
		{
		    if(j%i==0)
			a[j]=!a[j];   //非0即真，这里真假值，即1和0两种情况。	
		}	
	} 
	//输出开着的灯编号。
	for(int i=1;i<=n;i++)
	{
	   if(a[i])
	   {
	   	if(first)        //使两个数之间有空格。 
	   	first=0;
	   	else
	   	{
	   	  cout<<" ";	
		}
		cout<<i;
	   }	
	} 
   	return 0;
}