清华复试机试之求root(N,k)

题目描述

    N<k时，root(N,k) = N，否则，root(N,k) = root(N',k)。N'为N的k进制表示的各位数字之和。输入x,y,k，输出root(x^y,k)的值 (这里^为乘方，不是异或)，2=<k<=16，0<x,y<2000000000，有一半的测试点里 x^y 会溢出int的范围(>=2000000000) 

输入描述:

    每组测试数据包括一行，x(0<x<2000000000), y(0<y<2000000000), k(2<=k<=16)

输出描述:

    输入可能有多组数据，对于每一组数据，root(x^y, k)的值

示例1

输

4 4 10

输出

4

分析：本题在设计时，注意目的时为了考察对于可能超出int型的大数计算次方问题的处理，本题的题眼在于最终比较的是root的两个输入大小的比较，即N的计算只要在k的范围内即可，只需要与k比较大小，因此可以用k来mod，简化计算，并且数据的范围可以控制在int的范围内，值得注意的是，在测试过程中，发现只有55%的数据可以case，最终通过比较学习AC过的的大牛的代码，才发现自己的程序不仅繁琐，而且应该在自定义函数FastAB的首句下面加上 a%=mod;只有这样当a输入的很大时，也不会第一次计算N就会溢出出现与答案不同的结果

 本题还仿照《王道机试指南》上的类似题目学会了利用二进制化进行减小计算量

附上牛客网OJ网址

https://www.nowcoder.com/practice/9324a1458c564c4b9c4bfc3867a2aa66?tpId=60&tqId=29488&tPage=1&ru=/kaoyan/retest/1001&qru=/ta/tsing-kaoyan/question-ranking

#include<iostream>
#include<cstdio>
#include<string>

using namespace std;

int FastAB(int a,int b,int mod){
    int ans=1;//保存最后结果
    a%=mod;//先降低
    while(b!=0){
        //转成二进制
        if(b%2==1){  //若为奇数，则乘上自己的a倍，类似与 2^7=2^6*2
            ans=(ans*a)%mod;
        }
        b/=2;
        a=(a*a)%mod;//a进行平方
         //去模进行降低计算

    }    return ans;

}

int main(){
    int a,b,k;
    while(scanf("%d%d%d",&a,&b,&k)!=EOF){

       int s = FastAB(a,b,k-1);
       if(s==0) cout<<k-1<<endl;
        else cout<<s<<endl;

    }
}