开灯问题

开灯问题

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

#include <iostream>
using namespace std;

int main()
{
    int a[1000];
    int n,k;
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        a[i]=-1;
    }
    for(int i=1;i<=k;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(j%i==0)
            {
                a[j]=(-1)*a[j];
            }
        }
    }
    for(int i=1;i<=n;i++)
    {
        if(a[i]==1)
        {
            cout<<i<<" ";
        }
    }
    return 0;
}