FZU 2018 计数 快速幂取模

题意：对于方程 x^x = a(mod p), PH想知道对于[0,p-1]内的数，有多少个这样的x满足这个方程。请注意，虽然对于0^0的值有争论，甚至不一定有意义，可是在本题中，PH认为0^0 = 1。

#include <iostream>#include <cstdio>#include <cstring>using namespace std;const int MAXN = 1010;int p;char a[MAXN];int result(int a,int b,int m){    long long d,t;    d=1;    t=a;    while (b>0)    {        if (b%2==1)            d=(d*t)%m;        b/=2;        t=(t*t)%m;    }    return (int)d;}int main(){    int t;    scanf("%d%*c",&t);    while (t--){        scanf("%s %d",a,&p);        int ans = 0;        for (int i = 0; i < strlen(a); i++)            ans = (ans*10+(a[i]-'0')) % p;        if (strcmp(a,"1") == 0 && p == 1)            ans = 1;        int cnt = 0;        for (int i = 1; i < p; i++){            int sum = result(i,i,p);            if (sum == ans)                cnt++;          }        if (ans == 1)            cnt++;         printf("%d\n",cnt);    }    return 0;}