【作业帮】算法笔试第一题

if __name__ == "__main__":
    x, n = [int(x) for x in input().strip().split(' ')]
#    for i in range(1, n+1):
#        res += pow(x, i)
#    print(res%998244353)
    if x == 1:
        print(n % 998244353)
    else:
        res = (pow(x, n+1) - x)//(x-1)
        print(res % 998244353)
一直 过不了 算法 复杂度过大,在本地 确实 如果 n, x 取7位数以上 运行会超过2秒#笔试题目##作业帮##算法工程师#
全部评论
这种暴力算法不如C++啊
点赞 回复
分享
发布于 2019-08-14 20:50
等比数列求和加快速幂
点赞 回复
分享
发布于 2019-08-15 06:18
小红书
校招火热招聘中
官网直投
貌似python过不了
点赞 回复
分享
发布于 2019-08-19 11:57

相关推荐

点赞 2 评论
分享
牛客网
牛客企业服务