NYOJ 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

注-此题为：  NYOJ 77 开灯问题

已AC代码：

#include<cstdio>
int main()
{
int n,k,i,j,s[1100]; // s 的值 表示 灯的 状态
scanf("%d%d",&n,&k);
for(i=1;i<=n;++i)
s[i]=0; //为 0 是 关着的 ，为 1 开着
for(i=1;i<=k;++i) // k 个人
{
for(j=1;j<=n;++j) // n 个灯
{
if(j%i==0) // 是否为 i 的倍数
{
if(s[j]==0) //把 关着 的 打开
s[j]=1;
else if(s[j]==1) // 或 把 开着的 关掉
s[j]=0;
}
}
}
int flag=1; // 输出 格式控制 标志
for(i=1;i<=n;++i)
{
if(s[i])
{
if(flag) //最后无空格
printf("%d",i);
else
printf(" %d",i);
flag=0;
}
}
return 0;
}