开灯问题
时间限制:
3000 ms | 内存限制:
65535 KB
难度:
1
-
描述
-
有n盏灯,编号为1~n,第1个人把所有灯打开,第2个人按下所有编号为2 的倍数的开关(这些灯将被关掉),第3 个人按下所有编号为3的倍数的开关(其中关掉的灯将被打开,开着的灯将被关闭),依此类推。一共有k个人,问最后有哪些灯开着?输入:n和k,输出开着的灯编号。k≤n≤1000
-
输入
- 输入一组数据:n和k 输出
- 输出开着的灯编号 样例输入
-
7 3
样例输出
-
1 5 6 7
我的代码:
#include<cstdio>
#include<iostream>
using namespace std;
int main()
{
int n,k,i,j;
cin>>n>>k;;
bool a[1001]={0};
for(i=2; i<=k; i++)
{
for(j=1; j*i<=n; j++)
a[i*j] = !a[i*j];
}
for(i=1; i<=n; i++)
if(0 == a[i])
cout<<" "<<i;
cout<<endl;
return 0;
}
最优代码:
#include <iostream>
using namespace std;
int main()
{
int n,k,a[1000],i;
cin>>n>>k;
for(i=0;i<n;i++)
a[i]=1;
for(i=2;i<=k;i++)
for(int j=0;j<n;j++)
{
if((j+1)%i==0)
if(a[j]==0)a[j]=1;
else a[j]=0;
}
for(i=0;i<n;i++)
if(a[i]==1)cout<<i+1<<" ";
cout<<endl;
return 0;
}