开关灯 C++题解

开关灯

内存限制： 256 MiB 时间限制： 1000 ms 标准输入输出 题目类型： 传统 评测方式： 文本比较

题目描述

假设有 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 <bits/stdc++.h>
using namespace std;
const int maxn = 5e3 + 5;
int n, m, x, sum;
int a[maxn];
int main() {
	scanf("%d %d", &n, &m);
    for(int i = 1; i <= n; i++) {
    	a[i] = -1;
    	for(int j = 1; j <= m; j++) {
	    	if(i % j == 0) {
	    		a[i] *= -1;
			} 
		} 
	} 
	for(int i = 1; i <= n; i++) {
    	if(a[i] == 1) {
    		sum++;
		} 
	} 
	for(int i = 1; i <= n; i++) {
    	if(a[i] == 1) {
    		x++;
    		if(x != sum) {
	    		printf("%d,", i);
			} else {
				printf("%d", i);
			} 
		} 
	} 
	return 0;
}