Kotlin Kata - 尾遞迴版級數平方和
輸入正整數 n,回傳 1^2 + 2^2 + … + n^2 的總和
題目保證輸入的 n > 0 且 n < 999_999
題目要求必須使用尾遞迴的方式撰寫
Example 1:
Input: 10
Output: 55
,1^2 + 2^2 + … + 10^2 = 385
Example 2:
Input: 1
Output: 1
tailrec fun squareSum(n: BigInteger): BigInteger{}
解答
點擊展開解答
要用尾遞迴的方式處理費式數列
squareSum()
函數的參數是不夠的
我們必須要宣告新的函數來進行處理
利用 BigInteger 的操作
加上利用預設參數,我們可以寫成
tailrec fun squareSum(number: BigInteger, answer: BigInteger = BigInteger.ZERO): BigInteger {
return when (number) {
BigInteger.ZERO -> answer
else -> sum(number - BigInteger.ONE, answer + number * number)
}
}