P1683 狱吏问题

题目描述

某国王大赦囚犯，让以狱吏 n 次通过一排锁着的 n 间牢房并规定：狱吏每通过一次，就转动一次这排牢房中的某些门锁。门锁每转动一次，原来锁着的就被打开，原来打开的就被锁上。狱吏通过 n 次后，门锁为开的牢房中的犯人被放出，门锁为锁上的牢房中的犯人不得获释。转动门锁的规则如下：狱吏第一次通过牢房，要转动每一把门锁，即全部锁打开；第二次通过牢房时，从第二间开始转动，每隔一间转动一次；第 k 次通过牢房时，从第 k 间开始转动，每隔 k-1 间转动一次。问：狱吏通过 n 次后，哪些牢房的锁是打开的？

输入描述

多组输入，输入一个正整数 n（n<1000000），表示牢房的间数，牢房编号从 1 开始。

输出描述

输出锁是开着的牢房的编号（每个编号之间用一个空格分开）。

样例输入

Copy to Clipboard

15 20 

样例输出

Copy to Clipboard

1 4 9 1 4 9 16

 

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int n;
    while(cin>>n)
    {
        for(int i=1;i<=n;i++)
        {
            if(sqrt(i)==(int)sqrt(i))
                cout<<i<<" ";
        }
        cout<<endl;
    }
}