开灯问题

有n个灯，k个人，第一个人把所有灯都打开，第二个人按下所有编号为2的倍数的开关（这些灯将被关掉），第三个人按下编号是3的倍数的开关（其中关掉的灯将2被打开，开着的灯将关闭），一次类推

样例输入：

7 3

样例输出：

1 5 6 7

#include<stdio.h>
#include<string.h>
#define maxn 1010
int a[maxn];
int main()
{
int n,k,first=1;
memset(a,0,sizeof(a));//初始化数组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];//就是给a[j]赋与其相反的值真假值，例如，如果a[j] = 5, 那这个赋值后，a[j] = 0如果，a[j]=0,赋值语句后，a[j]=1；

for(int i=1;i<=n;i++)
if(a[i])
{
if(first)//控制输出格式
first=0;
else
printf(" ");
printf("%d",i);
}
printf("\n");
return 0;
}