算法竞赛入门经典 开灯问题C++解法

开灯问题

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

样例输入：

7 3

样例输出：

1 5 6 7

#include<iostream>
#include<string.h>
#define imax 1010
using namespace std;
int a[imax