Add Kotlin code for computational complexity (#1090)
* feat(kotlin):new kotlin support files * fix(kotlin): reviewed the formatting, comments and so on. * fix(kotlin): fix the indentation and format * feat(kotlin): Add kotlin code for the backtraking chapter. * fix(kotlin): fix incorrect output of preorder_traversal_iii_template.kt file * fix(kotlin): simplify kotlin codes * fix(kotlin): modify n_queens.kt for consistency. * feat(kotlin): add kotlin code for computational complexity. * fix(kotlin): remove iteration folder. * fix(kotlin): remove n_queens.kt file out of folder. * fix(kotlin): remove some folders. * style(kotlin): modified two chapters.pull/1096/head^2
parent
92f82cbcca
commit
306dc019ef
@ -0,0 +1,74 @@
|
|||||||
|
/**
|
||||||
|
* File: iteration.kt
|
||||||
|
* Created Time: 2024-01-25
|
||||||
|
* Author: curtishd (1023632660@qq.com)
|
||||||
|
*/
|
||||||
|
|
||||||
|
package chapter_computational_complexity.iteration
|
||||||
|
|
||||||
|
/* for 循环 */
|
||||||
|
fun forLoop(n: Int): Int {
|
||||||
|
var res = 0
|
||||||
|
// 循环求和 1, 2, ..., n-1, n
|
||||||
|
for (i in 1..n) {
|
||||||
|
res += i
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
|
||||||
|
/* while 循环 */
|
||||||
|
fun whileLoop(n: Int): Int {
|
||||||
|
var res = 0
|
||||||
|
var i = 1 // 初始化条件变量
|
||||||
|
// 循环求和 1, 2, ..., n-1, n
|
||||||
|
while (i <= n) {
|
||||||
|
res += i
|
||||||
|
i++ // 更新条件变量
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
|
||||||
|
/* while 循环(两次更新) */
|
||||||
|
fun whileLoopII(n: Int): Int {
|
||||||
|
var res = 0
|
||||||
|
var i = 1 // 初始化条件变量
|
||||||
|
// 循环求和 1, 4, 10, ...
|
||||||
|
while (i <= n) {
|
||||||
|
res += i
|
||||||
|
// 更新条件变量
|
||||||
|
i++
|
||||||
|
i *= 2
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 双层 for 循环 */
|
||||||
|
fun nestedForLoop(n: Int): String {
|
||||||
|
val res = StringBuilder()
|
||||||
|
// 循环 i = 1, 2, ..., n-1, n
|
||||||
|
for (i in 1..n) {
|
||||||
|
// 循环 j = 1, 2, ..., n-1, n
|
||||||
|
for (j in 1..n) {
|
||||||
|
res.append(" ($i, $j), ")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return res.toString()
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Driver Code */
|
||||||
|
fun main() {
|
||||||
|
val n = 5
|
||||||
|
var res: Int
|
||||||
|
|
||||||
|
res = forLoop(n)
|
||||||
|
println("\nfor 循环的求和结果 res = $res")
|
||||||
|
|
||||||
|
res = whileLoop(n)
|
||||||
|
println("\nwhile 循环的求和结果 res = $res")
|
||||||
|
|
||||||
|
res = whileLoopII(n)
|
||||||
|
println("\nwhile 循环 (两次更新) 求和结果 res = $res")
|
||||||
|
|
||||||
|
val resStr = nestedForLoop(n)
|
||||||
|
println("\n双层 for 循环的遍历结果 $resStr")
|
||||||
|
}
|
@ -0,0 +1,76 @@
|
|||||||
|
/**
|
||||||
|
* File: recursion.kt
|
||||||
|
* Created Time: 2024-01-25
|
||||||
|
* Author: curtishd (1023632660@qq.com)
|
||||||
|
*/
|
||||||
|
|
||||||
|
package chapter_computational_complexity.recursion
|
||||||
|
|
||||||
|
import java.util.*
|
||||||
|
|
||||||
|
/* 递归 */
|
||||||
|
fun recur(n: Int): Int {
|
||||||
|
// 终止条件
|
||||||
|
if (n == 1)
|
||||||
|
return 1
|
||||||
|
// 递: 递归调用
|
||||||
|
val res = recur(n - 1)
|
||||||
|
// 归: 返回结果
|
||||||
|
return n + res
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 使用迭代模拟递归 */
|
||||||
|
fun forLoopRecur(n: Int): Int {
|
||||||
|
// 使用一个显式的栈来模拟系统调用栈
|
||||||
|
val stack = Stack<Int>()
|
||||||
|
var res = 0
|
||||||
|
// 递: 递归调用
|
||||||
|
for (i in n downTo 0) {
|
||||||
|
stack.push(i)
|
||||||
|
}
|
||||||
|
// 归: 返回结果
|
||||||
|
while (stack.isNotEmpty()) {
|
||||||
|
// 通过“出栈操作”模拟“归”
|
||||||
|
res += stack.pop()
|
||||||
|
}
|
||||||
|
// res = 1+2+3+...+n
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Kotlin tailrec 关键词使函数实现尾递归优化 */
|
||||||
|
tailrec fun tailRecur(n: Int, res: Int): Int {
|
||||||
|
// 终止条件
|
||||||
|
if (n == 0)
|
||||||
|
return res
|
||||||
|
// 尾递归调用
|
||||||
|
return tailRecur(n - 1, res + n)
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 斐波那契数列:递归 */
|
||||||
|
fun fib(n: Int): Int {
|
||||||
|
// 终止条件 f(1) = 0, f(2) = 1
|
||||||
|
if (n == 1 || n == 2)
|
||||||
|
return n - 1
|
||||||
|
// 递归调用 f(n) = f(n-1) + f(n-2)
|
||||||
|
val res = fib(n - 1) + fib(n - 2)
|
||||||
|
// 返回结果 f(n)
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Driver Code */
|
||||||
|
fun main() {
|
||||||
|
val n = 5
|
||||||
|
var res: Int
|
||||||
|
|
||||||
|
res = recur(n)
|
||||||
|
println("\n递归函数的求和结果 res = $res")
|
||||||
|
|
||||||
|
res = forLoopRecur(n)
|
||||||
|
println("\n使用迭代模拟递归求和结果 res = $res")
|
||||||
|
|
||||||
|
res = tailRecur(n, 0)
|
||||||
|
println("\n尾递归函数的求和结果 res = $res")
|
||||||
|
|
||||||
|
res = fib(n)
|
||||||
|
println("\n斐波那契数列的第 $n 项为 $res")
|
||||||
|
}
|
@ -0,0 +1,109 @@
|
|||||||
|
/**
|
||||||
|
* File: space_complexity.kt
|
||||||
|
* Created Time: 2024-01-25
|
||||||
|
* Author: curtishd (1023632660@qq.com)
|
||||||
|
*/
|
||||||
|
|
||||||
|
package chapter_computational_complexity.space_complexity
|
||||||
|
|
||||||
|
import utils.ListNode
|
||||||
|
import utils.TreeNode
|
||||||
|
import utils.printTree
|
||||||
|
|
||||||
|
/* 函数 */
|
||||||
|
fun function(): Int {
|
||||||
|
// 执行某些操作
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 常数阶 */
|
||||||
|
fun constant(n: Int) {
|
||||||
|
// 常量、变量、对象占用 O(1) 空间
|
||||||
|
val a = 0
|
||||||
|
var b = 0
|
||||||
|
val nums = Array(10000) { 0 }
|
||||||
|
val node = ListNode(0)
|
||||||
|
// 循环中的变量占用 O(1) 空间
|
||||||
|
for (i in 0..<n) {
|
||||||
|
val c = 0
|
||||||
|
}
|
||||||
|
// 循环中的函数占用 O(1) 空间
|
||||||
|
for (i in 0..<n) {
|
||||||
|
function()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 线性阶 */
|
||||||
|
fun linear(n: Int) {
|
||||||
|
// 长度为 n 的数组占用 O(n) 空间
|
||||||
|
val nums = Array(n) { 0 }
|
||||||
|
// 长度为 n 的列表占用 O(n) 空间
|
||||||
|
val nodes = mutableListOf<ListNode>()
|
||||||
|
for (i in 0..<n) {
|
||||||
|
nodes.add(ListNode(i))
|
||||||
|
}
|
||||||
|
// 长度为 n 的哈希表占用 O(n) 空间
|
||||||
|
val map = mutableMapOf<Int, String>()
|
||||||
|
for (i in 0..<n) {
|
||||||
|
map[i] = i.toString()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 线性阶(递归实现) */
|
||||||
|
fun linearRecur(n: Int) {
|
||||||
|
println("递归 n = $n")
|
||||||
|
if (n == 1)
|
||||||
|
return
|
||||||
|
linearRecur(n - 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 平方阶 */
|
||||||
|
fun quadratic(n: Int) {
|
||||||
|
// 矩阵占用 O(n^2) 空间
|
||||||
|
val numMatrix: Array<Array<Int>?> = arrayOfNulls(n)
|
||||||
|
// 二维列表占用 O(n^2) 空间
|
||||||
|
val numList: MutableList<MutableList<Int>> = arrayListOf()
|
||||||
|
for (i in 0..<n) {
|
||||||
|
val tmp = mutableListOf<Int>()
|
||||||
|
for (j in 0..<n) {
|
||||||
|
tmp.add(0)
|
||||||
|
}
|
||||||
|
numList.add(tmp)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 平方阶(递归实现) */
|
||||||
|
tailrec fun quadraticRecur(n: Int): Int {
|
||||||
|
if (n <= 0)
|
||||||
|
return 0
|
||||||
|
// 数组 nums 长度为 n, n-1, ..., 2, 1
|
||||||
|
val nums = Array(n) { 0 }
|
||||||
|
println("递归 n = $n 中的 nums 长度 = ${nums.size}")
|
||||||
|
return quadraticRecur(n - 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 指数阶(建立满二叉树) */
|
||||||
|
fun buildTree(n: Int): TreeNode? {
|
||||||
|
if (n == 0)
|
||||||
|
return null
|
||||||
|
val root = TreeNode(0)
|
||||||
|
root.left = buildTree(n - 1)
|
||||||
|
root.right = buildTree(n - 1)
|
||||||
|
return root
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Driver Code */
|
||||||
|
fun main() {
|
||||||
|
val n = 5
|
||||||
|
// 常数阶
|
||||||
|
constant(n)
|
||||||
|
// 线性阶
|
||||||
|
linear(n)
|
||||||
|
linearRecur(n)
|
||||||
|
// 平方阶
|
||||||
|
quadratic(n)
|
||||||
|
quadraticRecur(n)
|
||||||
|
// 指数阶
|
||||||
|
val root: TreeNode? = buildTree(n)
|
||||||
|
printTree(root)
|
||||||
|
}
|
@ -0,0 +1,166 @@
|
|||||||
|
/**
|
||||||
|
* File: time_complexity.kt
|
||||||
|
* Created Time: 2024-01-25
|
||||||
|
* Author: curtishd (1023632660@qq.com)
|
||||||
|
*/
|
||||||
|
|
||||||
|
package chapter_computational_complexity.time_complexity
|
||||||
|
|
||||||
|
/* 常数阶 */
|
||||||
|
fun constant(n: Int): Int {
|
||||||
|
var count = 0
|
||||||
|
val size = 10_0000
|
||||||
|
for (i in 0..<size)
|
||||||
|
count++
|
||||||
|
return count
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 线性阶 */
|
||||||
|
fun linear(n: Int): Int {
|
||||||
|
var count = 0
|
||||||
|
// 循环次数与数组长度成正比
|
||||||
|
for (i in 0..<n)
|
||||||
|
count++
|
||||||
|
return count
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 线性阶(遍历数组) */
|
||||||
|
fun arrayTraversal(nums: IntArray): Int {
|
||||||
|
var count = 0
|
||||||
|
// 循环次数与数组长度成正比
|
||||||
|
for (num in nums) {
|
||||||
|
count++
|
||||||
|
}
|
||||||
|
return count
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 平方阶 */
|
||||||
|
fun quadratic(n: Int): Int {
|
||||||
|
var count = 0
|
||||||
|
// 循环次数与数组长度成平方关系
|
||||||
|
for (i in 0..<n) {
|
||||||
|
for (j in 0..<n) {
|
||||||
|
count++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return count
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 平方阶(冒泡排序) */
|
||||||
|
fun bubbleSort(nums: IntArray): Int {
|
||||||
|
var count = 0
|
||||||
|
// 外循环:未排序区间为 [0, i]
|
||||||
|
for (i in nums.size - 1 downTo 1) {
|
||||||
|
// 内循环:将未排序区间 [0, i] 中的最大元素交换至该区间的最右端
|
||||||
|
for (j in 0..<i) {
|
||||||
|
if (nums[j] > nums[j + 1]) {
|
||||||
|
// 交换 nums[j] 与 nums[j + 1]
|
||||||
|
nums[j] = nums[j + 1].also { nums[j + 1] = nums[j] }
|
||||||
|
count += 3 // 元素交换包含 3 个单元操作
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return count
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 指数阶(循环实现) */
|
||||||
|
fun exponential(n: Int): Int {
|
||||||
|
var count = 0
|
||||||
|
// 细胞每轮一分为二,形成数列 1, 2, 4, 8, ..., 2^(n-1)
|
||||||
|
var base = 1
|
||||||
|
for (i in 0..<n) {
|
||||||
|
for (j in 0..<base) {
|
||||||
|
count++
|
||||||
|
}
|
||||||
|
base *= 2
|
||||||
|
}
|
||||||
|
// count = 1 + 2 + 4 + 8 + .. + 2^(n-1) = 2^n - 1
|
||||||
|
return count
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 指数阶(递归实现) */
|
||||||
|
fun expRecur(n: Int): Int {
|
||||||
|
if (n == 1) {
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
return expRecur(n - 1) + expRecur(n - 1) + 1
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 对数阶(循环实现) */
|
||||||
|
fun logarithmic(n: Float): Int {
|
||||||
|
var n1 = n
|
||||||
|
var count = 0
|
||||||
|
while (n1 > 1) {
|
||||||
|
n1 /= 2
|
||||||
|
count++
|
||||||
|
}
|
||||||
|
return count
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 对数阶(递归实现) */
|
||||||
|
fun logRecur(n: Float): Int {
|
||||||
|
if (n <= 1)
|
||||||
|
return 0
|
||||||
|
return logRecur(n / 2) + 1
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 线性对数阶 */
|
||||||
|
fun linearLogRecur(n: Float): Int {
|
||||||
|
if (n <= 1)
|
||||||
|
return 1
|
||||||
|
var count = linearLogRecur(n / 2) + linearLogRecur(n / 2)
|
||||||
|
for (i in 0..<n.toInt()) {
|
||||||
|
count++
|
||||||
|
}
|
||||||
|
return count
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 阶乘阶(递归实现) */
|
||||||
|
fun factorialRecur(n: Int): Int {
|
||||||
|
if (n == 0)
|
||||||
|
return 1
|
||||||
|
var count = 0
|
||||||
|
// 从 1 个分裂出 n 个
|
||||||
|
for (i in 0..<n) {
|
||||||
|
count += factorialRecur(n - 1)
|
||||||
|
}
|
||||||
|
return count
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Driver Code */
|
||||||
|
fun main() {
|
||||||
|
// 可以修改 n 运行,体会一下各种复杂度的操作数量变化趋势
|
||||||
|
val n = 8
|
||||||
|
println("输入数据大小 n = $n")
|
||||||
|
|
||||||
|
var count: Int = constant(n)
|
||||||
|
println("常数阶的操作数量 = $count")
|
||||||
|
|
||||||
|
count = linear(n)
|
||||||
|
println("线性阶的操作数量 = $count")
|
||||||
|
count = arrayTraversal(IntArray(n))
|
||||||
|
println("线性阶(遍历数组)的操作数量 = $count")
|
||||||
|
|
||||||
|
count = quadratic(n)
|
||||||
|
println("平方阶的操作数量 = $count")
|
||||||
|
val nums = IntArray(n)
|
||||||
|
for (i in 0..<n) nums[i] = n - i // [n,n-1,...,2,1]
|
||||||
|
count = bubbleSort(nums)
|
||||||
|
println("平方阶(冒泡排序)的操作数量 = $count")
|
||||||
|
|
||||||
|
count = exponential(n)
|
||||||
|
println("指数阶(循环实现)的操作数量 = $count")
|
||||||
|
count = expRecur(n)
|
||||||
|
println("指数阶(递归实现)的操作数量 = $count")
|
||||||
|
|
||||||
|
count = logarithmic(n.toFloat())
|
||||||
|
println("对数阶(循环实现)的操作数量 = $count")
|
||||||
|
count = logRecur(n.toFloat())
|
||||||
|
println("对数阶(递归实现)的操作数量 = $count")
|
||||||
|
|
||||||
|
count = linearLogRecur(n.toFloat())
|
||||||
|
println("线性对数阶(递归实现)的操作数量 = $count")
|
||||||
|
|
||||||
|
count = factorialRecur(n)
|
||||||
|
println("阶乘阶(递归实现)的操作数量 = $count")
|
||||||
|
}
|
@ -0,0 +1,47 @@
|
|||||||
|
/**
|
||||||
|
* File: worst_best_time_complexity.kt
|
||||||
|
* Created Time: 2024-01-25
|
||||||
|
* Author: curtishd (1023632660@qq.com)
|
||||||
|
*/
|
||||||
|
|
||||||
|
package chapter_computational_complexity.worst_best_time_complexity
|
||||||
|
|
||||||
|
/* 生成一个数组,元素为 { 1, 2, ..., n },顺序被打乱 */
|
||||||
|
fun randomNumbers(n: Int): Array<Int?> {
|
||||||
|
val nums = IntArray(n)
|
||||||
|
// 生成数组 nums = { 1, 2, 3, ..., n }
|
||||||
|
for (i in 0..<n) {
|
||||||
|
nums[i] = i + 1
|
||||||
|
}
|
||||||
|
// 随机打乱数组元素
|
||||||
|
val mutableList = nums.toMutableList()
|
||||||
|
mutableList.shuffle()
|
||||||
|
// Integer[] -> int[]
|
||||||
|
val res = arrayOfNulls<Int>(n)
|
||||||
|
for (i in 0..<n) {
|
||||||
|
res[i] = mutableList[i]
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 查找数组 nums 中数字 1 所在索引 */
|
||||||
|
fun findOne(nums: Array<Int?>): Int {
|
||||||
|
for (i in nums.indices) {
|
||||||
|
// 当元素 1 在数组头部时,达到最佳时间复杂度 O(1)
|
||||||
|
// 当元素 1 在数组尾部时,达到最差时间复杂度 O(n)
|
||||||
|
if (nums[i] == 1)
|
||||||
|
return i
|
||||||
|
}
|
||||||
|
return -1
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Driver Code */
|
||||||
|
fun main() {
|
||||||
|
for (i in 0..9) {
|
||||||
|
val n = 100
|
||||||
|
val nums: Array<Int?> = randomNumbers(n)
|
||||||
|
val index: Int = findOne(nums)
|
||||||
|
println("\n数组 [ 1, 2, ..., n ] 被打乱后 = ${nums.contentToString()}")
|
||||||
|
println("数字 1 的索引为 $index")
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in new issue