南阳oj_(77)开灯问题

开灯问题

时间限制：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;
}