求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

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

//根据Root(a*b,k)=Root(Root(a,k)*Root(b,k),k)这个规律，可以把幂次方变成类似于二分求幂，从而不会溢出

int root(int num, int k)  //将十进制数转换为k进制后，返回k进制各数字之和
{
	while (num >= k)
	{
		int tran = 0;
		while (num != 0)
		{
			tran += num % k;
			num /= k;
		}
		num = tran;
	}
	return num;
}

int trans(int y, int b[50])  //将y变成二进制数
{
    int i = 0;
    while (y > 0)
    {
        b[i++] = y % 2;
        y /= 2;
    }
    int len = i;  //转化为二进制之后的长度
    return len;
}
int main()
{
    int x, y, k, b[50];
    while(cin >> x >> y >> k)
    {
	    int ans0 = root(x, k);
        int ans = 1;
        int len = trans(y, b);
        for (int i = len - 1; i >= 0; i--)
        {
            ans = root(ans*ans, k);
            if (b[i] == 1)
                ans = root(ans*ans0, k);
        }
        cout << ans << endl;
    }
    return 0;
}

//更简便的方法
链接：https://www.nowcoder.com/questionTerminal/9324a1458c564c4b9c4bfc3867a2aa66
来源：牛客网

#include <iostream>
using namespace std;
int main(int argc,char* argv[])
{
    int x,y,k;
    while(cin>>x>>y>>k)
    {
        int ans = 1; 
        x = x % (k - 1); 
        while(y > 0) 
        { 
            if(y % 2 == 1) 
                ans = (ans * x) % (k - 1); 
            y = y >> 1; 
            x = (x * x) % (k - 1); 
        }
        cout<<(ans == 0 ? k - 1 : ans)<<endl;
    }
    return 0;
}