开关灯（c语言）

题目描述

假设有N盏灯(N为不大于5000的正整数)，从1到N按顺序依次编号，初始时全部处于开启状态；有M个人(M为不大于N的正整数)也从1到M依次编号。

第一个人(1号)将灯全部关闭，第二个人(2号)将编号为2的倍数的灯打开，第三个人(3号)将编号为3的倍数的灯做相反处理（即将打开的灯关闭，将关闭的灯打开）。依照编号递增顺序，以后的人都和3号一样，将凡是自己编号倍数的灯做相反处理。

请问：当第M个人操作之后，哪几盏灯是关闭的，按从小到大输出其编号，其间用逗号间隔。

输入

输入正整数N和M，以单个空格隔开。

输出

顺次输出关闭的灯的编号，其间用逗号间隔。

样例

输入数据 1

10 10

输出数据 1

1,4,9

我的代码：

#include<stdio.h>
#include<stdbool.h>

int main(){
    int n,m;
    scanf("%d %d",&n,&m);
    int i;
    if(m==1){ //只有一个人直接输出所有
        for(i=0;i<n-1;i++){
            printf("%d,",i+1);
        }
        printf("%d",n);
        return 0;
    }else{
        bool light[n]; //记录灯开/关
        for(i=0;i<=n;i++){
            light[i]=false;  //初始化灯，默认第一个人已经关闭全部灯。
        }
        int j=1;
        while(j!=m){  //依次遍历m个人操作灯
            j++;
            for(i=1;i<=n;i++){
                if(i%j==0){
                    light[i]=!light[i];  //为倍数时，取反。
                }
            }
        }
        int flag=0;
        for(i=1;i<=n;i++){
            if(light[i]==false&&flag==0){  //格式问题，第一个不输出“,”
                flag=1;
                printf("%d",i);
            }else if(light[i]==false){
                printf(",%d",i);  //依次输出灯关闭的下标
            }
        }
    }
    return 0;
}