算法篇-开灯问题

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

样例输入：7 3

样例输出：1 5 6 7

 

代码如下：

#include <iostream>
#include <cstring>
#define MAXN 1001                //数组开大点
using namespace std;
int a[MAXN];
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])
        {
            if(first)
                first=0;
            else
                cout<<" ";
            cout<<i;
        }
    return 0;
}

运行结果：

 

题目感觉还是有点难度的，至少我看见题目第一眼没想出怎么写好，看了书上的c语言代码才有了想法，从这个题目中学到了两点：

1：！不是取负，而是将0变成1或1变成0（这是我自己认为的，百度搜不到，也不知道是不是正确的）。以前一直以为就是取负。

2：可以通过设置标志变量来避免输出多余空格。如题，第一个i前面是没有空格的，当输出第一个i后再输出i前会输出一个空格。