水仙花数

#include <iostream>
using namespace std;

int main()
{
    int n;
    cin >> n;
    int cnt = 0,num = 0,sum = 0,a = n,s = 0;
    while(n != 0)
    {
        s++;
        if(s == 1)
        {
            cnt = n % 10;
        }
        else if(s == 2)
        {
            num = n % 10;
        }
        else
        {
            sum = n % 10;
        }
        n /= 10;
    }
    if(cnt * cnt * cnt + num * num * num + sum * sum * sum == a)
    {
        cout << "YES";
    }
    else
    {
        cout << "NO";
    }
    return 0;
}