POJ31:开关灯

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

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

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

输入 输入正整数N和M，以单个空格隔开。 输出 顺次输出关闭的灯的编号，其间用逗号间隔。

#include<iostream>
#include<stdio.h>
#include<math.h>
using namespace std;

int main()
{
	int N, M, light[5001] = {0},d=0;//0表示开，1关
	cin >> N >> M;
	for (int i = 1; i <= M; i++)//M人
	{
		for (int j = i; j <= N; j = j + i)//N灯
		{
			if (light[j] == 0)light[j] = 1;//另一种方法：设灯初始值为1，1表示开，-1关，每次变化*-1
			else light[j] = 0;
		}
	}
	for (int j = 1; j <= N; j ++)
	{
		if (light[j] == 1)
		{
			if (d == 0)//第一个前面没有逗号
			{
				cout << j ;
				d = 1;
			}				
			else//其他元素前面都有,
			{
				cout << "," << j;
			}
		}
	}
	return 0;
}