【2020秋招笔试】百信银行:快速计算阶乘+快速幂

O ( l o g n ) O(logn) O(logn)时间复杂度的情况下计算 t ! t ! t! a t a ^ t at
快速幂计算在笔试中写过好多次了,万万没想到阶乘也可以。

import sys
def power(n, m):
    global number, cn, p_size, mask
    if (m == 1):
        cur = n
    elif(m & 1 != 0):
        cur = n * power(n, m - 1)
    else:
        cur = power(n, m >> 1)
        cur *= cur
        cn[number] = (cur >> ((m >> 1) * p_size)) & mask  
        number += 1
    return cur

def factor(n):
    global number, cn, p_size, mask
    if (n == 1):
        return 1
    elif(n & 1 == 1):
        return n * factor(n - 1)
    else:
        cur = factor(n >> 1)
        pos = number
        number += 1
        return cn[pos] * cur * cur
    
def factorial(n):
    global number, cn, p_size, mask
    x = (1 << n) + 1
    number = 0
    mask = (1 << n) - 1
    p_size = n
    power(x, n)
    number = 0
    return factor(n)
p_size = 0
mask = 0
cn = []
number = 0
N = int(sys.stdin.readline().strip())
for ppp in range(N):
    line = sys.stdin.readline().strip()
    a,t = list(map(int, line.split()))
    cn = [0 for i in range(t+1)]
    p = power(a, t)
    number = 0
    q = factorial(t)
    print(max(p, q))
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值