HDU-2035人见人爱A^B

HDU-2035人见人爱A^B
题目：
求A^B的最后三位数表示的整数。
说明：A^B的含义是“A的B次方”
Input
输入数据包含多个测试实例，每个实例占一行，由两个正整数A和B组成（1<=A,B<=10000），如果A=0, B=0，则表示输入数据的结束，不做处理。
Output
对于每个测试实例，请输出A^B的最后三位表示的整数，每个输出占一行。
Sample Input
2 3
12 6
6789 10000
0 0
Sample Output
8
984
1

#include<iostream>
#include<cstdio>
using namespace std;

int main()
{
    int n,m,sum;
    while(scanf("%d %d",&n,&m) != EOF)
    {
        sum = 1;
        if(!n && !m) break;
        for(int i = 0;i < m;i++)
        {
            sum = sum * n;
            sum = sum % 1000;
        }
        cout << sum % 1000 << endl;
    }
    return 0;
}

一开始这道题想通过for循环来计算A^B，然后把计算出来的值对1000取余就是最后三位了。但是，看到例子中的6789的10000次方，就打消了这样的想法。因为题目中只要求最后三位的数字，所以在for循环中每次累乘之后要不停的对算出来的值对1000取余，那么这样就可以控住数的范围了。
但是此时的复杂度还是o(n)，如何才能更快速呢？查阅资料之后发现有一种快速幂取模法，此时算法复杂度可缩减为o(logn)级别，例如：12^6 = 12^2* 12^4

代码如下：

#include<stdio.h>
int mod_pow(int x, int n,int mod){      //快速幂
    int res = 1;
    while( n > 0 ){
        if( n % 2 ) res = res * x % mod;//判断次方的奇偶
        x = x * x % mod;
        n >>= 1;
    }
    return res;
}
int main(){ 
    int m,n;
    while(scanf("%d%d",&m,&n),n||m)
        printf("%d\n",mod_pow(m,n,1000));
    return 0;
}

运行结果：