算法竞赛入门经典 程序3-2 开灯问题

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

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define maxn 1005
int a[maxn];
int main(){
    int n, k;     //n盏灯，k个人
    int first = 1;
    memset(a, 0, sizeof(a));
    scanf_s(