You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
hello-algo/codes/kotlin/chapter_divide_and_conquer/binary_search_recur.kt

49 lines
1.1 KiB

/**
* File: binary_search_recur.kt
* Created Time: 2024-01-25
* Author: curtishd (1023632660@qq.com)
*/
package chapter_divide_and_conquer.binary_search_recur
/* 二分查找:问题 f(i, j) */
fun dfs(
nums: IntArray,
target: Int,
i: Int,
j: Int
): Int {
// 若区间为空,代表无目标元素,则返回 -1
if (i > j) {
return -1
}
// 计算中点索引 m
val m = (i + j) / 2
return if (nums[m] < target) {
// 递归子问题 f(m+1, j)
dfs(nums, target, m + 1, j)
} else if (nums[m] > target) {
// 递归子问题 f(i, m-1)
dfs(nums, target, i, m - 1)
} else {
// 找到目标元素,返回其索引
m
}
}
/* 二分查找 */
fun binarySearch(nums: IntArray, target: Int): Int {
val n = nums.size
// 求解问题 f(0, n-1)
return dfs(nums, target, 0, n - 1)
}
/* Driver Code */
fun main() {
val target = 6
val nums = intArrayOf(1, 3, 6, 8, 12, 15, 23, 26, 31, 35)
// 二分查找(双闭区间)
val index = binarySearch(nums, target)
println("目标元素 6 的索引 = $index")
}