开关灯

题目描述

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

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

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

输入格式

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

输出格式

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

输入样例 

10 10

输出样例 

1,4,9

【代码示例】

 

#include<iostream>
using namespace std;
// 数组初始化为0，表示灯全部开启。 
int a[10000]={}, b[10000]={}; 

int main(){
    int n, m, flag=0, cnt=0;
    cin>>n>>m;
    // 表示n个人。 
    for(int i=1; i<=m; i++){
         // 每个人按灯。
         for(int j=1; j<=n; j++){
             if(j%i==0) {
                 a[j] = 1-a[j];
             }
         } 
    } 
    // 去掉逗号：统计亮灯的数目以便于计算逗号的数目。 
    for(int i=1; i<=n; i++){
        if(a[i]==1) b[cnt++]=i;
    }
    // 输出前cnt-1个亮灯序号，最后输出最后一个亮灯的序号。 
    for(int i=0; i<cnt; i++){
        if(i<cnt-1) cout<<b[i]<<",";
        else if(i==cnt-1) cout<<b[i];
    }
    return 0;
}