3-2例题-开灯问题

开灯问题。

有n盏灯，编号为1~n。第一个人把所有灯打开，第二个人按下所有编号为2的倍数的开关（这些灯将被关掉），第三个人按下所有编号为3的倍数的开关（其中关掉的灯将被打开，开着的灯将被关闭），依此类推，一共有K个人，问最后有哪些灯开着？输出开着的灯的编号。k<=n<=1000。

#include<cstdio>
#include<string.h>
#define maxn 1010
int a[maxn];
using namespace std;
int main()
{
  int n,k,first=1;
  scanf("%d %d",&n,&k);
  memset(a,0,sizeof(a));
  for(int i=1;i<=k;i++)
  for(int j=1;j<=n;j++)
  if(j%i==0)a[j]=!a[j];
  for(int i=1;i<=n;i++)
  if(a[i])
  {
    if(first)first=0;
    else
    printf(" ");
    printf("%d",i);
  }
  printf("\n");
  return 0;
}

【分析】

用a[1],a[2],...,a[n]表示为编号为1，2，3，...n的灯是否开着。模拟这些操作即可