算法竞赛_开灯问题_C++

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

样例输入：7 3

样例输出：1 5 6 7

#include <bits/stdc++.h>#define maxN 1000+10using namespace std;int a[maxN];int main(){    int i,j,k,n,first =1;    memset(a, 0, sizeof(a));    cin >> 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(first)                first = 0;            else                cout << " ";            cout << i;        }    }    cout << endl;return 0;}

说明：
memset()函数的作用是将数组a复制为0，标志变量first控制输出空格。

**参考《算法竞赛入门经典_刘汝佳著》**