CCF NOI1008. 水仙花数 (C++)

1008. 水仙花数

题目描述

输入一个三位数n，判断是否为水仙花数，如果是则输出“YES”，不是则输出“NO”。水仙花数：是指一个3位数，它的每个位上的数字的3次幂之和等于它本身。（例如：1³ + 5³+ 3³ = 153）

输入

输入一个三位数n。

输出

按照题目描述输出对应结果。

样例输入

153

样例输出

YES

数据范围限制

100<=n<=999

C++代码

#include <iostream>
#include <cmath>
#include <cassert>

using namespace std;

int main()
{
    int n;

    cin >> n;

    assert(n>=100 && n<=999);

    int tmp = n;
    int sum = 0;

    while(tmp > 0)
    {
        sum += (int)pow(tmp%10, 3.0);
        tmp /= 10;
    }
    
    if (sum == n)
    {
        cout << "YES" << endl;
    }
    else
    {
        cout << "NO" << endl;
    }

    return 0;
}