C语言 开灯问题

开灯问题。有n盏灯,编号为1~n。第1个人把所有灯打开,第2个人按下所有编号为2的倍数的开关(这些灯将被关掉),第3个人按下所有编号为3的倍数的开关(其中关掉的灯将被打开,开着的灯将被关闭),依此类推。一共有k个人,问最后有哪些灯开着?输入n和k,输出开着的灯的编号。k≤n≤1000。

样例输入:
7 3

样例输出:
1 5 6 7

---

代码如下：

#include <stdio.h>
#define MAXN 1005
int lamps[MAXN];

int main()
{
    int n, k;
    while(2 == scanf("%d%d", &n, &k))
    {
        int i, j;
        for(i = 1; i <= n; ++i)
            lamps[i] = 1;
        for(i = 2; i <= k; ++i)
            for(j = 1; j * i <= n; ++j)
                lamps[i*j] ^= 1;
        for(i = 1; i <= n; ++i)
            if(lamps[i]) printf("%d ", i);
        putchar('\n');
    }
    return 0;
}

看了别人的代码，感觉受益匪浅：

#include <stdio.h>
#include <string.h>
#define MAXN 1005
int lamps[MAXN];

int main()
{
    int n, k;
    while(2 == scanf("%d%d", &n, &k))
    {
        memset(lamps, 0, sizeof(lamps));
        int i, j;
        for(i = 1; i <= k; ++i)
            for(j = 1; j <= n; ++j)
                if(j % i == 0) lamps[j] ^= 1;
        for(i = 1; i <= n; ++i)
            if(lamps[i]) printf("%d ", i);
        putchar('\n');
    }
}