【作业帮】算法笔试第一题
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秒#笔试题目##作业帮##算法工程师#