算法入门-开灯问题

有n盏灯，编号为1～n。第1个人把所有灯打开，第2个人按下所有编号为2第倍数的开关，第3个人按下所有编号为3的倍数的开关（其中关掉的灯被打开，打开的灯被关掉），以此类推，一共有k个人，问最后有哪些灯开着。(k <= n <= 1000)

//
// Created by yoyoOvQ on 2021/1/25.
//

#include <stdio.h>
#include <string.h>
#define maxn 1005

int a[maxn];

int main(){
    int n,k;
    int t = 1;  // 控制空格
    scanf("%d%d",&n,&k);

    memset(a,0,sizeof(a));

    for (int i = 1; i <= k; ++i) {
        for (int j = 1; j <= n; ++j) {
            if (j%i==0)
                a[j] = !a[j];
        }
    }
    for (int i = 1; i <= n; ++i) {
        if (a[i]){
            if (t == 1){
                t = 0;
            }else{
                printf(" ");
            }
            printf("%d",i);
        }
    }
    return 0;
}