【水题】开灯问题

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

样例输入：7 3

样例输出：1 5 6 7

#include <stdio.h>
#include <string.h>
#define MAXN 1500
int a[MAXN];
int main()
{

    int n,k;
    int first=1; //标志变量
    while(scanf("%d%d",&n,&k)!=EOF)  //依次输入灯的数量，人的数量。
    {
        for(int i=1;i<=k;i++)   //当第i个人进行操作
        {
            for(int j=1;j<=n;j++) //用循环模拟第n盏灯的情况
            {
                if(j%i==0) a[j]=!a[j];  // a[j]为1表示灯亮，0表示灯灭
            }
        }
        for(int j=1;j<=n;j++)
        {
            if(a[j])
            {
                if(first) first =0;  //第一个数不要空格，最后一个数之后也不要空格
                else printf(" ");
                printf("%d",j);
            }
        }
        printf("\n");
        memset(a,0,sizeof(a)); //变量初始化
        first=1;
    }
    return 0;
}