清华大学2010年机试-求root(N, k)-1085

最新推荐文章于 2020-02-17 12:16:33 发布

Elias777

最新推荐文章于 2020-02-17 12:16:33 发布

阅读量323

点赞数

分类专栏：九度OJ 文章标签：九度OJ

本文链接：https://blog.csdn.net/Elias777/article/details/77309282

版权

九度OJ 专栏收录该内容

51 篇文章 0 订阅

订阅专栏

清华大学2010年机试-求root(N, k)-1085

题目描述：
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)的值
样例输入
4 4 10
样例输出
4

解题思路：
首先，我能想到的最笨的方法就是先做幂运算，计算出N之后，再计算root(N, k)，毫无疑问，这个方法的时间复杂度过高。然后我想到了快速幂，结果还是错误。这应该是由于输入的数据过大，甚至超过了long long的存储范围。查阅网上资料得知，该题使用快速幂取模的方法做。（果然对模运算掌握的还是太差）
本题可以有如下分析：
N=a0+a1*k+a2*k^2+……+an*k^n;
N’=a0+a1+a2+……+an;
N-N’=a1*(k-1)+a2*(k^2-1)+……+an*(k^n-1);
提取(k-1)有: (N-N’)%(K-1)=0;
继续递推下去有：（N-N’）%(k-1) =0；
（N’-N”）%(k-1)=0;
……
（N(r-1)-N(r)）%（k-1）=0；
相加有：（N-N(r)）%(k-1)=0,N(r)是我们要求的结果，故有
N(r) = N % (k-1);
如果 N(r)==0 ,为边界，则 N(r) = k-1;

#include <cstdio>
typedef long long ll;
int x, y, k;

ll powMod(int x, int y, int k){
    ll res = 1, base = x, mod = k-1;
    while(y != 0){
        if(y&1 != 0) res = (res * base) % mod;
        base = (base * base) % mod;
        y >>= 1;
    }
    return res;
}

int main(){
    while(scanf("%d%d%d", &x, &y, &k) != EOF){
        ll N = powMod(x, y, k);
        if (N == 0) N = k-1;
        printf("%lld\n", N);
    }
    return 0;
}