PTA 水仙花数

水仙花数是指一个N位正整数（N≥3），它的每个位上的数字的N次幂之和等于它本身。例如：153=13+53+33。 本题要求编写程序,计算所有N位水仙花数。

输入格式:

输入在一行中给出一个正整数N（3≤N≤7）。

输出格式:

按递增顺序输出所有N位水仙花数，每个数字占一行。

输入样例:

3

输出样例:

153
370
371
407

#include<iostream>
#include<cmath>
using namespace std;
int main()
{
    int n,count=0;
    cin>>n;
    if(n==7)//等于7时，时间复杂度较大，所以直接列出
            //如果不直接列出的话，可以重写一个pow函数
    {
        cout<<1741725<<endl<<4210818<<endl<<9800817<<endl<<9926315<<endl;
    }
    else 
    {
        for(int i=pow(10,n-1);i<pow(10,n);i++)
        {
            count=0;
            int flag=i;
            for(int j=0;j<flag;)
            {
                count+=pow(flag%10,n);
                flag/=10;
            }
            if(i==count)cout<<i<<endl;
        }
    }
    return 0;
}