两有序数组求第K大的数字/k大的数字/快排
思路: 二分+递归
对k二分,分别比较a,b k/2位置的大小,小的那一部分就可以删除了,再更新k
basecase是 k为1或者 其中一个数组删完了
package main
import "fmt"
func findKth(a, b []int, aleft, bleft, k int) int {
// a 为空,直接返回b的下标
if aleft >= len(a) {
return b[bleft+k-1]
}
// b 为空,直接返回a的下标
if bleft >= len(b) {
return a[aleft+k-1]
}
// k 为 1 即开头哪个更小
if k == 1 {
return min(a[aleft], b[bleft])
}
// 二分k 同事保证ab数组不会越界
mid := k / 2
aidx := min(aleft+mid-1, len(a)-1)
bidx := min(bleft+mid-1, len(b)-1)
// a 的 idx 小于b的,那么a的idx之前的肯定不是, k减去这一段,继续寻找
// b同理
if a[aidx] < b[bidx] {
return findKth(a, b, aidx+1, bleft, k-(aidx-aleft+1))
}
return findKth(a, b, aleft, bidx+1, k-(bidx-bleft+1))
}
func min(a, b int) int {
if a > b {
return b
}
return a
}
func main() {
aa := []int{2, 3, 4}
bb := []int{1}
fmt.Println(findKth(aa, bb, 0, 0, 4))
}
快排/求数组第k大数字
package main
import (
"fmt"
"math/rand"
"time"
)
func shuffle(a []int, l, r int) int{
idx := rand.Intn(r-l+1) + l
a[r], a[idx] = a[idx], a[r]
return partition(a,l,r)
}
// 1,4,2,1,3
// 1,2,4,1,3
// 1,2,1,4,3
// 1,2,1,3,4
func partition(a []int, l, r int) int {
pivot := a[r]
i := l - 1
for j := l; j < r; j++ {
if a[j] >= pivot {
i++
a[i], a[j] = a[j], a[i]
}
}
a[i+1], a[r] = a[r], a[i+1]
return i + 1
}
func findKth(a []int, l, r, k int) int {
cur := partition(a, l, r)
//fmt.Println(a)
if cur == k-1 {
return a[k-1]
} else if cur < k {
return findKth(a, cur+1, r, k)
} else {
return findKth(a, l, cur-1, k)
}
}
func qqsort(a []int, l, r int) {
if l < r {
idx := partition(a, l, r)
qqsort(a, l, idx-1)
qqsort(a, idx+1, r)
}
}
func main() {
rand.Seed(time.Now().UnixNano())
a := []int{1, 2, 3, 4, 5}
fmt.Println(a)
fmt.Println(findKth(a, 0, len(a)-1, 1))
}
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 在鹅厂做java开发是什么体验
· 百万级群聊的设计实践
· WPF到Web的无缝过渡:英雄联盟客户端的OpenSilver迁移实战
· 永远不要相信用户的输入:从 SQL 注入攻防看输入验证的重要性
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析