【编程题练习】牛客-阿里：小强爱数学

最新推荐文章于 2024-03-23 12:00:47 发布

Jared�

最新推荐文章于 2024-03-23 12:00:47 发布

阅读量789

点赞数

分类专栏：编程题

本文链接：https://blog.csdn.net/Jaredzzz/article/details/119384822

版权

编程题专栏收录该内容

7 篇文章 2 订阅

订阅专栏

小强发现当已知x+y=A以及xy=B时,能很轻易的算出的x^2 + y^2值.但小强想请你在已知A和B的情况下,计算出x^n + y^n的值.因为这个结果可能很大,所以所有的运算都在模1e9+7下进行.

输入描述:
第一行输入一个正整数T.表示有T组数据接下来T行,每行输入三个整数A,B和n.

输出描述:
输出T行,每一行表示每组数据的结果.

示例1
输入
3
4 4 3
2 3 4
5 2 6
输出
16
999999993
9009

'''
感觉是一个递归问题，数学问题
'''
mod = int(1e9+7)
def calculate(A,B,N):
    if N == 1:
        return A
    elif N == 2:
        return (A**2-2*B)%mod
    else:
        if N%2==0:
            AnBn = 1
            for i in range(N//2):
                AnBn = AnBn*B%mod
            AnplusBn = calculate(A,B,N//2)
            return (AnplusBn * AnplusBn - 2 * AnBn) % mod
        else:
            AnBn = 1
            for i in range(N//2):
                AnBn = AnBn*B%mod
            return (calculate(A,B,N//2)*calculate(A,B,N//2+1) - A * AnBn) % mod

T = int(input())
for _ in range(T):
    inp = input().split(" ")
    A = int(inp[0]) % mod        
    B = int(inp[1]) % mod
    N = int(inp[2])
    print(int(calculate(A, B, N)))