PTA—开灯问题

题目描述

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

输入格式

n和k，输出开着的灯编号。k≤n≤1000。

输出格式

输出开着的灯编号

输入样例

7  3

样例输出

1  5  6  7

#include <stdio.h>
#include <string.h>
#define MAXN 1000+10
int main()
{
    int n,k,first=1,a[MAXN]; 
    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]==1)
	    {
	        if(first)
			    first=0;
		    else
			    printf(" ");
		    printf("%d",i);
	    }
	}
    printf("\n");     
    return 0;
}