开灯问题——基础

描述

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

输入

输入一组数据：n 和 k

输出

输出开着的灯编号

样例输入

7 3
样例输出

1 5 6 7
对于开关灯，我们使用0表示灯暗，1表示灯亮。

思路看下图

代码实现：

public static void main(String[] args){
	    int a=10;//灯的数量
	    int b=4;//人数
	    int[] sum=new int[100];
	    for(int i=1;i<=a;i++){
	        for(int j=1;j<=b;j++){
	            if(i%j==0){
	                if(sum[i]==0){
	                    sum[i]=1;
                    }else {
	                    sum[i]=0;
                    }

                }
            }
        }
        for (int i=1;i<=a;i++){
            if (sum[i]==1) {
                System.out.print(i+" ");
            }
        }
    }