1209: 【一维数组】开灯问题（lights）

题目描述

    有n盏灯，编号为1~n。第1上人把所有灯打开，第2个人按下所有编号为2的倍数的开关（这些灯将被关掉），第3个人按下所有编号为3的倍数的开关（其中关掉的灯将打开，开着的灯将被关闭），依此类推。一共有k个人，问最后有哪些灯开着？输入：n和k，输出开着的灯的编号。k=<n<=1000。

输入

两个数n和k。

输出

开着的灯的编号。

样例输入

复制

7 3

样例输出

复制

1 5 6 7

点个赞吧，还有关注！！！
上代码！！！

#include <bits/stdc++.h>

using namespace std;

#define N 1005

int x[N];

int main() {

    int n, k,i;

    cin >> n >> k;

    for (i = 1; i <= k; i++) {

        for (int j = 1; j <= n; j++) {

            if (j % i == 0) {

                if (x[j] == 0)

                   x[j] = 1; else

                                                    x[j] = 0;

            }

        }

    }

    for (i = 1; i <= n; i++)

        if (x[i] == 1) {

        cout << i;

        break;

    }

    for (i=i+1; i <= n; i++)

        if (x[i] == 1)

        cout <<" "<< i;

    return 0;

}