【华为练习题 】 n位水仙花数（初级）

【华为练习题 】 n位水仙花数（初级）

题目

水仙花数又称阿姆斯特朗数。
水仙花数是指一个n 位数( n≥3 )，它的每个位上的数字的n 次幂之和等于它本身。（例如：1^3 + 5^3 + 3^3 = 153）
求输入的数字是否为水仙花数

解答

#include <iostream>
#include <vector>
using namespace std;

bool isRight(int n){
    vector<int> v;
    int sum = 0, tmp = n;
    while (tmp)
    {
        v.push_back(tmp % 10);
        tmp /= 10;
    }   
    for (unsigned i = 0; i < v.size(); i++)
    {
        sum += (int)pow(v[i],v.size());
    }
    return n == sum;
}

int main()
{
    int n;
    cin >> n;
    cout << (isRight(n) ? "Yes" : "No") << endl;
    return 0;
}