UVA 12895 - Armstrong Number Solution

problem link-click here


#include<bits/stdc++.h>
using namespace std;

int power(int x,int y)
{
    int sum=x;

    for(int i=1;i<y;i++)
    {
        sum=sum*x;
    }

    return sum;
}

int main()
{
    char str[20];

    int n;

    cin>>n;

    while(n--)
    {
        cin>>str;
        long long int sum=0,s;
        s=atoi(str);

        for(int i=0;i<strlen(str);i++)
        {
            sum=sum+power((str[i]-48),strlen(str));
        }

        if(sum==s)
        {
            cout<<"Armstrong"<<endl;
        }
        else
        {
            cout<<"Not Armstrong"<<endl;
        }
    }

    return 0;
}