Virtusa Interview Question

Write a function to identify Armstrong number