开灯问题

问题描述

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

实现代码

#include<stdio.h>
#include<string.h>
#define maxn 101
int a[maxn];
int main(){
    int n, k, first = 1;
    memset(a, 0, sizeof(a));
    scanf("%d%d", &n, &k);
    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(first){
                first = 0;
            }else{
                printf(" ");
            }
            printf("%d", i);
        }
    }
    printf("\n");
    return 0;
}

memset(a, 0, sizeof(a));的作用试把数组a清零，它在string.h中定义。虽然也能用for循环完成相同的任务，但是用memset又方便又快捷。另一个技巧在输出：为了避免输出多余空格，设置了一个标志变量first，可以表示当前要输出的变量是否为第一个。第一个变量前不应有空格，但其他变量都有。