PTA 浙大版《C语言程序设计（第4版）》题目集 习题4-6 水仙花数

原题链接

问题描述

水仙花数是指一个N位正整数（N≥3），它的每个位上的数字的N次幂之和等于它本身。例如：153=1³+5³+3³。 本题要求编写程序,计算所有N位水仙花数。

输入格式:
输入在一行中给出一个正整数N（3≤N≤7）。

输出格式:
按递增顺序输出所有N位水仙花数，每个数字占一行。

输入样例:
3

输出样例:
153
370
371
407

代码长度限制
16 KB
时间限制
2500 ms
内存限制
64 MB

解析

本题逻辑难度很小，但是有一点需要注意，盲目暴力求解可能会运行超时，主要问题出在库中的pow()函数上。该函数的返回值为double类型，两个参数也为double，而我们需要传入的参数为int类型，返回值也为int类型，调用该函数会被强制转换成double类型，而double类型数据的运算会比int类型数据的运算复杂些，这也是造成运行超时的原因。改进方法为自己写一个int pow(int,int)函数切结幂运算。

代码

#include 

int pow(int a,int b)
{
	int res=1;
	for(int i=1;i<=b;i++) res*=a;
	return res;
}

int judge(int n)
{
	int sum=0,count=0;
	int n1=n,n2=n;
	do
	{
		count++;
		n1/=10;	
	}while(n1);
	while(n2)
	{
		sum+=pow(n2%10,count);
		n2/=10;
	}
	if(sum==n) return 1;
	return 0;
}

int main()
{
	int n;
	scanf("%d",&n);
	for(int i=pow(10,n-1);i<pow(10,n);i++)
	{
		if(judge(i)) printf("%d\n",i);
	}
	return 0;	
} 

运行结果