1

第三章 算法竞赛入门经典

3-2 开灯问题

有n盏灯,编号为1~n。第1个人把所有灯打开,第2个人按下所有比编号为2的倍数的开关(这些灯将被关掉),第3个人按下所有编号为3的倍数的开关(其中关掉的灯江北打开,开着的灯将被关闭),以此类推,一共有k个人,问最后有哪些等开着?输入nk,输出开着的灯的编号。k≤n≤1000

样例输入:

7 3

样例输出:

1 5 6 7
#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;        
}

黄奋斗
163 声望19 粉丝