开灯问题

开灯问题

时间限制：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

 

这个问题不难，算是作为好久没有做题又回归战线的练习了。可以让人很快进入状态。开始的时候忘记了用0和1来标记灯的亮灭，忽略了第一个输出变量前不能有空格。后来改正之后设置了标志变量first。  这两个点算是解这个题的关键呗~~

 

代码如下：

#include <iostream>
#include <cstring>
using namespace std;
int a[1000];
int main()
{
    int i,j,n,k,first=1;
    memset(a,0,sizeof(a));
    cin>>n>>k;
    for(i=1;i<=k;i++)
    {
        for(j=1;j<=n;j++)
        {
            if(j%i==0)
            a[j]=!a[j];
        }
    }
    for(i=1;i<=n;i++)
    {
        if(a[i]==1)
        {
            if(first==1)
            {
                cout<<i;
                first=0;
            }

            else
            {
                cout<<" "<<i;
            }
        }
    }
    cout<<endl;
    return 0;
}