nyoj-开灯问题

/*开灯问题
时间限制：3000 ms | 内存限制：65535 KB
难度：1
描述
有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>
#include<string.h>
#define M 1100
int a[M];
int main()
{
int i,j,k,n,first=1;
memset(a,0,sizeof(a));//把数组a清零
scanf("%d%d",&n,&k);
for(i=1;i<=k;i++)
for(j=1;j<=n;j++)
if(j%i==0) a[j]=!a[j];
for(i=1;i<=n;i++)
if(a[i])
{
if(first) first=0;
else printf(" ");
printf("%d",i);
}
printf("\n");
return 0;
}