C语言每日小练（二）——开灯问题

例：开灯问题

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

样例输入：7 3

样例输出：1 5 6 7

解：开一个足够大的数组，模拟整个过程即可~

Code：

#include<stdio.h>
#include<string.h>

#define MAX 1000+10
int a[MAX];

int main()
{
    int i, j, n, k;
    memset(a, 0, sizeof(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(i == n) printf("%d\n", i);
        else printf("%d ", i);
    }
    return 0;
}

运行结果：