信息学奥赛一本通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>
using namespace std;

//1109:
int main(){
	int a[5001];
	int N,M;
	cin>>N>>M;
	for(int i=1;i<=N;i++)//因为题目中标记1开始，为避免混淆，值的记录从a[1]开始，后同;
		a[i]=1;
	for(int i=1;i<=M;i++)
		for(int j=1;i*j<=N;j++)//j为j倍，i*j即为i的j倍;
			a[i*j]=a[i*j]*-1;
	int j=1;//设置一个为1的标记;
	for(int i=1;i<=N;i++)
		if(a[i]==-1&&j==1){
			cout<<i;
			j=0;//当第一盏关闭的灯被记录后，标记改为0;
		}
		else if(a[i]==-1&&j==0)//因为标记j==0，所以这些都不是第一盏灭了的灯;
			cout<<","<<i;//循环输出数值前加一个","即可;
	return 0;
}