HDOJ 1097 A hard puzzle

//HDOJ 1097 A hard puzzle
//二分快速求幂
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<ctype.h>
int main()
{
    int cmp(int a,int b);
    int a,b;
    int s;
    while(~scanf("%d%d",&a,&b))
    {
        s=cmp(a,b);
        printf("%d\n",s);
    }
    return 0;
}
int cmp(int a,int b)
{
    int s=1;
    a%=10;//初始化a只保留个位
    while(b)
    {
        if(b&1)
        {
        s*=a;
        s%=10;//s保留个位
        }
        b>>=1;
        a*=a;
        a%=10;//a保留个位
    }
    return s;
}