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 1005int 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 1005int 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');    }}