编程题-水仙花数

水仙花数

• 内存限制:64MB 时间限制:1000ms 特判: No

• 
  
• 难度:0

题目描述:

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

输入描述:

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

输出描述:

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

样例输入:

153 154 0

样例输出:

Yes
No

代码：

#include <stdio.h>

int count(int num){
	int num_a;
	int num_b;
	int num_c;
	int sum = 0;
	num_a = num;
	for(int i = 0 ; i < 3 ; i++){
		num_b = num_a % 10;
		num_c = num_b;
		num_b = num_b * num_b;
		num_c = num_b * num_c;
		sum += num_c; 
		num_a = num_a / 10;
	}
	return sum;
}

int main(){
	int n;
	int c;
	while(1){
		scanf("%d",&n);
		if(n == 0){
			break;
		}else if(n < 100){
			printf("No\n");
		}else if(n >= 1000){
			printf("No\n");
		}else{
			if(n==count(n)){
				printf("Yes\n");
			}else{
				printf("No\n");
			}
		}
	}
	return 0;
}