判断一个数是否为水仙花数

判断一个数是否为"水仙花数"，所谓"水仙花数"是指这样的一个数：首先是一个三位数，其次，其各位数字的立方和等于该数本身。例如：371是一个"水仙花数"，371=3^3+7^3+1^3.

#include<stdio.h>
#include<math.h>
int main()
{   int D,a[3],i,s=0;
    scanf("%d",&D);
    for(i=0;i<3;i++)
    {
        a[i]=D%10;
        D=D/10;
        s=s+pow(a[i],3);
    }
       if(s==a[0]+a[1]*10+a[2]*100)
         printf("1\n");
       else printf("0\n");
    return 0;
}