程序3-2 开灯问题

程序3-2 开灯问题

有n盏灯，编号为1-n，第一个人把所有灯打开，第二个人按下所有编号两倍的开关（这些灯被关掉），第三个人按下所有编号三倍的开关（关掉的灯会被打开，开着的会被关掉），依此类推，一共有k个人，问最后有哪些灯开着？

输入：n和k，输出开着的灯的编号，k<=n<=1000

样例输入：

7 3 

样例输出：

1 5 6 7

---

分析：用数组a【】标记灯是否开着。用for嵌套实现每个人的开关灯。

学到：1.用menset(a,0,sizeof(a))代替 for语句 初始化a【】值都为0；

          2.定义一个变量first标志输出是否为第一个，第一个前面没有空格，其它都有；

          3.用两个for循环解决每次开关灯

书本源代码

#include<stdio.h>
#include<string.h>

#define maxn 1010

int a[maxn];
int main()
{
	int n,k,first=1;
	//for(int s=0;s<maxn;s++)
	//a[s]=0;
	memset(a,0,sizeof(a));
	scanf("%d%d",&n,&k);
	for(int i=1;i<=k;i++)
		for(int j=1;j<=n;j++)
			if(j%i==0)a[j]=!a[j];
		for(int q=1;q<=n;q++)
			if(a[q])
			{
				if(first) first=0;
				else printf(" ");
				printf("%d",q);
			}
			printf("\n");
			return 0;
}