欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

算法竞赛_开灯问题_C++

程序员文章站 2022-05-09 21:18:21
有n盏灯,编号为1~n。第1个人把所有灯打开,第2个人按下所有编号为2的倍数的开关(这些灯将被关掉),第2个人按下所有的编号为3的倍数的开关(其中关掉的灯将被打开,开着的灯将被关闭...

有n盏灯,编号为1~n。第1个人把所有灯打开,第2个人按下所有编号为2的倍数的开关(这些灯将被关掉),第2个人按下所有的编号为3的倍数的开关(其中关掉的灯将被打开,开着的灯将被关闭),以此类推。一共有k个人,问最后有哪些灯开着?输入n和k,输出开着的灯编号。k ≤ n ≤ 1000。

样例输入:7 3

样例输出:1 5 6 7

#include 

#define maxN 1000+10

using namespace std;

int a[maxN];

int main(){
    int i,j,k,n,first =1;
    memset(a, 0, sizeof(a));
    cin >> n >> k;
    for(i=1; i<=k; i++)
        for(j=1; j<=n; j++)
            if(j%i == 0) a[j] = !a[j];
    for(i=1; i<=n; i++){
        if(a[i]){
            if(first)
                first = 0;
            else
                cout << " ";
            cout << i;
        }
    }
    cout << endl;
return 0;
}

说明:
memset()函数的作用是将数组a复制为0,标志变量first控制输出空格。