sicily 1454. Pseudoprime numbers

#include<iostream>    
#include<stdio.h>
#include<cmath>
using namespace std;
int is_prime(int x)
{
    int m=floor(sqrt((double)x)+0.5);
    for(int i=2;i<=m;++i)
        if(x%i==0)
            return 0;
    return 1;
}
int power(int a,int b,int m)    //快速幂取模计算 (a^b)%m
{
    if(a==0)
        return 0;
    else if(b==0)
        return 1;
    long long r=a%m,k=1;    
    while(b>1)
    {
        if((b&1)==1)
            k=(k*r)%m;    
        r=(r*r)%m;
        b=b>>1;
    }
    return (r*k)%m;
}
int main()
{
    int p,a;
    while(scanf("%d%d",&p,&a) && p )
    {
        if( !is_prime(p) && power(a,p,p)==a )
            printf("yes\n");
        else
            printf("no\n");
    }
    return 0;
}