JAVA版本开灯问题

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

C语言版本

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

JAVA版本

import java.util.Arrays;
import java.util.Scanner;

public class kaidengwenti {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		int n,k;		
		Scanner scanner= new Scanner(System.in);
		n=scanner.nextInt();
		k=scanner.nextInt();
		int[] a=new int[1010];
		Arrays.fill(a, 0);
		
		for(int i=1;i<=k;i++) {
			for(int j=1;j<=n;j++) {
				if(j%i==0) {
					if(a[j]==0) {
						a[j]=1;
					}else {
						a[j]=0;
					}
				}
			}
		}
		
		for(int i=1;i<=n;i++) {
			if(a[i]==1) {
					System.out.print(i+" ");
				
			}
		}
	}

}