南阳OJ_语言入门_77_开灯问题

开灯问题

时间限制： 3000 ms  |  内存限制： 65535 KB

难度： 1

描述

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

输入

输入一组数据：n和k

输出

输出开着的灯编号

样例输入

7 3

样例输出

1 5 6 7

#include<stdio.h>
#include<stdlib.h>
int main( void )
{

	int n = 0;
	int k = 0;
	int *a = NULL;
	int i = 0;
	int j = 2;
	scanf("%d%d",&n,&k);
	
	a = (int*)malloc(sizeof(int)*n+1);//分配数组
	
	for( i = 1; i <= n; i++ )//赋值为1，即初始为开灯状态
	{
		a[i] = 1;//1表示灯是打开状态
	}
	
//	for( i = 1; i <= n; i++ )
//	{
//		printf("%d ",a[i]);		
//	}
	
	for( j = 2; j <= k; j++ )
	{
		for( i = 1; i <= n; i++ )//如果灯的编号可以整除k，则按下开关
		{
			if( i % j == 0 )
			{
				a[i] = a[i] * -1;
			}			
		}
		
//		for( i = 1; i <= n; i++ )
//		{
//			printf("%d ",a[i]);		
//		}
//		
//		printf("\n");	
	}
	
	for( i = 1; i <= n; i++ )
	{
		if( 1 == a[i] )
		{
			printf("%d ",i);
		}
	}

	
	//printf("\n\nHello World!\n");
	return 0;
}