每日一点C语言--开灯问题

这里是《剑指offer》PDF下载链接，欢迎一起进步的同学下载学习！
剑指offer高清完整版PDF下载

开灯问题

有n盏灯，编号1-n， 第一个人将所有灯都打开，第二个人按下所有2的倍数的开关，第三个人按下所有3的倍数的开关，以此类推，一共k个人，问最后哪些灯是开着的？

#include 
#include 
#define maxn 1010
int a[maxn];           //较大的数组声明在函数之外
int main(){
    int i, j = 0;
    int n, k = 0;
    memset(a, 0, sizeof(a));
    scanf("%d%d", &n, &k);
    for(i = 1; i <= k; i++){
        for(j = i; j <= n; j++){
            a[j] = !a[j];
        }
    }
    for(i = 1; i <= n; i++){
        if(a[i]){
            printf("%d ", i);
        }
    }
    printf("\n");
    return 0;
}