首页 > 代码库 > 开灯问题--------《算法竞赛入门指导》P83

开灯问题--------《算法竞赛入门指导》P83

开灯问题。

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

 

我的代码为

#include<iostream>
using namespace std;
int main(void){
    int n,k;
    cin>>n; //灯的数量n 
    cin>>k; //人的数量k
    int i,j;
    int a[n];

    
    
    j=1;
    for(i=0;i<n;i++){
        a[i]=1;
    }
    for(j=2;j<=k;j++){
        for(i=j-1;i<n;i+=j){
            a[i]=-a[i];
        }    
    }
    
    for(i=0;i<n;i++){
        if(a[i]==1){
            cout<<i+1<<" ";
        }
    }
    return 0;
}

运行结果

技术分享

 

开灯问题--------《算法竞赛入门指导》P83