acm校庆数

#include<stdio.h>
#include<math.h>
#include<string.h>
int main ()
{
    int i = 0, j = 0, sum_1 = 0, sum_2 = 0;
    char a[30];
    scanf ( "%s", a );
    if ( ! ( strcmp ( a, "120") ) )
    {
        printf ( "YES\n" );
    }
        else
        {
            while ( a[i] != '\0' )
            {
                sum_1 += a[i] - '0';
                sum_2 += pow (  ( a[i] - '0' ), 2 );
                i ++;
            }
            if ( sum_1 == 120 || sum_2 == 120 )
            {
                printf ( "YES\n" );
            }
            else printf ( "NO\n" );
        }
    return 0;
}