開燈問題

描述

有n盞燈，編號為1~n，第1個人把所有燈開啟，第2個人按下所有編號為2 的倍數的開關（這些燈將被關掉），第3 個人按下所有編號為3的倍數的開關（其中關掉的燈將被開啟，開著的燈將被關閉），依此類推。一共有k個人，問最後有哪些燈開著？輸入：n和k，輸出開著的燈編號。k≤n≤1000

輸入

輸入一組資料：n和k

輸出

輸出開著的燈編號

樣例輸入

7 3

樣例輸出

1 5 6 7

思路：水題啊，無聊水一水，增加下人生的信心。

ac程式碼

#include<stdio.h>
#include<string.h>
#define MAX 100*100+10
int main(){
	int n,k;
	int a[MAX];
	scanf("%d%d",&n,&k);
	memset(a,0,sizeof(a));
	for(int i=1;i<=k;i++){
		for(int j=i;j<=n;j+=i){
			if(!a[j])
				a[j]=1;
			else
				a[j]=0;
		}
	}
	printf("%d",1);
	for(int i=2;i<=n;i++){
		if(a[i])
			printf(" %d",i);
	}
	printf("\n");
	return 0;
}