阿姆斯特朗数又称水仙花数,其判断方法为:一个n位数每一位的n次方之和若等于该数则是一个阿姆斯特朗数。
如153,是一个三位数;计算(1^3+5^3+3^3)=1+125+27=153,153是一个阿姆斯特朗数。
那么如果想判断并输出阿姆斯特朗数,首先需要对被判断n位数进行拆分,然后再计算各位数的n次方,最后相加。
我写了两个函数,时间复杂度较高,仅供参考,如有不足,还请指正(我才学编程一年,别喷)。
#include <bits/stdc++.h>
#pragma GCC optimize(0)
#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#pragma GCC optimize(4)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
ll n,a[1000005],wcl[100005],ycl[100005];
void split(ll x)
{
ll t=0;
while(x!=0)
{
wcl[++t]=x%10;
x/=10;
}
for(int i=1 ; i<=n ; i++)
{
ycl[i]=wcl[i];
for(int j=2 ; j<=n ; j++)ycl[i]*=wcl[i];
}
return;
}
void print(ll x)
{
ll sum=0;
for(int i=1 ; i<=n ; i++)
{
sum+=ycl[i];
}
if(sum==x)
{
cout<<sum<<endl;
}
return;
}
int main()
{
cin>>n;
ll k=1;
ll z=1;
for(int i=1 ; i<n ; i++)
{
k*=10;
}
z=k*10-1;
for(int i=k ; i<=z ; i++)
{
split(i);
print(i);
}
return 0;
}