今天被同学问到这样一个题目:
求A^B的最后三位数表示的整数。(1<=A,B<=10000)。如果直接求A^B次方的话,显然会超出数据类型的表示范围。但是仅仅作为后三位来说的话,最后三位的乘积实际上只跟上一次乘积的后三位有关,如(1234 * 1234) %1000 == (234 * 234 )%1000。因此整体代码如下:
int test(int A,int B)
{
int i,ret,tmp;
ret=tmp=A%1000;
for(i=1;i<B;i++)
{
ret *= (ret*tmp)%1000;
}
return ret;
}
但是这道题实际上,让我想到的是,在清华面试的时候,当时主考官给我出的一道算法题。求A^B次方对P求余,当然A,B的取值范围很大,也会发生数组越界的事情。其实这道题可以用数学证明如下:A^B%P求余的结果只能是两种结果,B取值为[1,无穷大],A^B%P的结果只有两种一种是{A%P,A^2%P, A^3%P,…0, 0,…}。即A^k 整除P。另一种情况是一个循环队列{A%P,A^2%P,A^3%P,…A^k%P,A^1%P,A^2%P}。
采用数学归纳法进行证明:
当B=P时,A^P%P = A%P成立。(A与B互质情况下,费马小定理)
假设A^N%P = A^k%P,那么只需证明A^(N+1) %P == A^(K+1) %P求余,则可以说明A^B %P求余数是一个循环队列。
因为设X = A^N %P = A^k%P, Y = A^(N+1)%P。
则A^N = mP + X; A^k = m'P + X; 则
A^(N+1)%P = A^N * A %P = (mP + X) * A % P = (m *A * P + X * A) % P = (X * A) % P;
A^(K+1)%P = A^K *A % P = (m'P + X) * A % P = (m' * A * P + X * A) %P = (X * A) % P;
因此 A^(N+1)%P = A^(K+1)%P 等式成立,所以A^B%P是一个循环队列,因此只要求出循环的前N个数就可以了。