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;	
}