南阳oj_(39)水仙花数

水仙花数

时间限制： 1000 ms  |  内存限制： 65535 KB 

难度： 0

描述 

请判断一个数是不是水仙花数。
其中水仙花数定义各个位数立方和等于它本身的三位数。

输入

有多组测试数据，每组测试数据以包含一个整数n(100<=n<1000)
输入0表示程序输入结束。

输出

如果n是水仙花数就输出Yes
否则输出No

样例输入

153
154
0

样例输出

Yes
No

我的代码：

#include<iostream>
using namespace std;
int main()
{
    int i,n,a,b,c;
    while(cin>>n)
{
    if(n!=0&&n>=100&&n<1000)
    {
    a=n/100;
    b=(n-a*100)/10;
    c=n-a*100-b*10;
    if(a*a*a+b*b*b+c*c*c==a*100+b*10+c)
        cout<<"YES"<<endl;
    else
        cout<<"NO"<<endl;
    }
    else
        break;
}
return 0;
}

最优代码：

 
#include<iostream>
using namespace std;
int main()
{
	int a;
	while(1)
	{
		cin>>a;
		if(a==0) break;
		cout<<((a==153||a==370||a==371||a==407)?"Yes":"No")<<endl;
	}
}