开关灯-题解

描述

赵老师的办公室有N盏灯(N为不大于5000的正整数)，从1到N按顺序依次编号，初始时全部处于开启状态；有M个卡皮巴拉(M为不大于N的正整数)也从1到M依次编号。

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

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

输入描述

输入正整数N和M，以单个空格隔开。
(N<=5000,M<=N)

输出描述

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

代码

#include <bits/stdc++.h>
using namespace std;

int main() {
    int n, m;
    cin >> n >> m;
    vector<int> ans;
    for (int i = 1; i <= n; i++) {
        bool tmp = false;
        for (int j = 1; j * j <= i; j++) {
            if (i % j == 0) {
                if (j <= m) {
                    tmp = !tmp;
                }
                int t = i / j;
                if (t != j && t <= m) {
                    tmp = !tmp;
                }
            }
        }
        if (tmp) {
            ans.push_back(i);
        }
    }
    for (int i = 0; i < ans.size(); i++) {
        if (i > 0) {
            cout << ",";
        }
        cout << ans[i];
    }
    return 0;
}