开灯问题（java实现）

问题：

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

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

样例输入：7 3
样例输出：1 5 6 7

思路：

• 利用数组进行标记，下标标记灯的编号，值标记灯的状态，-1为关闭，1为开启
• count记录灯的遍历情况，当count为k的倍数时，将灯的状态置反
• 最后输出值为1的灯

实现：

import java.util.Scanner;

public class Light {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner reader = new Scanner(System.in);
		System.out.println("请输入灯的数量");
		int n = reader.nextInt();
		System.out.println("请输入人的数量");
		int k = reader.nextInt();
		
		int a[] = new int[n+1];  
		for(int i=0;i<a.length;i++)
			a[i]=1;                 //1为打开状态,第一个人将灯全部打开
		int count=0;
		for(int i=2;i<=k;i++)
		{
			count=1;
			for(int j=1;j<=n;j++)
			{
				if(count%i==0)      //count为此时人的倍数时，将灯的状态置反
					a[j]=-a[j];
				count++;
			}
			/*测试中间状态
			for(int m=1;m<=n;m++)
			{
				if(a[m]==1)
					System.out.print(m+"  ");
			}
			System.out.println();*/
		}
		for(int i=1;i<=n;i++)
		{
			if(a[i]==1)
				System.out.print(i+"  ");
		}
	}
}