每日一题---372. 超级次方[力扣][Go]

题目描述

你的任务是计算 ab 对 1337 取模，a 是一个正整数，b 是一个非常大的正整数且会以数组形式给出。

解题代码

三叶大佬的思路：https://leetcode-cn.com/problems/super-pow/solution/gong-shui-san-xie-di-gui-kuai-su-mi-ying-yx1j/
我只是将三叶大佬的代码转换成为Go语言

var MOD = 1337
func superPow(a int, b []int) int {
	return dfs(a,len(b)-1,b)
}

func dfs(a,u int,b []int) int {
	if u == -1 {
		return 1
	}
	return qpow(dfs(a,u-1,b),10) * qpow(a,b[u]) % MOD
}

func qpow(a, b int) int {
	ans := 1
	a %= MOD
	for b != 0 {
		if (b & 1) !=0 {
			ans = ans*a %MOD
		}
		a = a*a%MOD
		b >>= 1
	}
	return ans
}

提交结果