一本通1109 開關燈

【題目描述】

假設有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>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
int main() {
    int n,m;
    cin>>n>>m;
    int a[n+1],b[n+1];
    for(int i=1; i<=n; i++) {
        a[i]=i;
        b[i]=1;//用1或-1來判斷開關
    }
    for(int j=1; j<=m; j++) {
        for(int i=1; i<=n; i++) {
            if(a[i]%j==0)
                b[i]=-b[i];
        }
    }
    cout<<1;
    for(int i=2; i<=n; i++) {
        if(b[i]==-1)
            cout<<","<<a[i];
    }

    return 0;
}