Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-02-04 06:54:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

优化二分查找算法

Browse Source
This commit is contained in:
orgin
2022-09-08 09:18:18 +08:00
parent 1520dae223
commit 280c04a5d7
1 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
util/algorithms/BiSearch.go
View File

@@ -10,7 +10,7 @@ type Element[ValueType NumberType] interface {
} }
//BiSearch 二分查找,切片必需有序号。matchUp表示是否向上范围查找。比如：数列10 20 30 ，当value传入25时，返回结果是2,表示落到3的范围 //BiSearch 二分查找,切片必需有序号。matchUp表示是否向上范围查找。比如：数列10 20 30 ，当value传入25时，返回结果是2,表示落到3的范围
func BiSearch[ValueType NumberType, T Element[ValueType]](sElement []T, value ValueType, matchUp bool) int { func BiSearch[ValueType NumberType, T Element[ValueType]](sElement []T, value ValueType, matchUp int) int {
low, high := 0, len(sElement)-1 low, high := 0, len(sElement)-1
if high == -1 { if high == -1 {
return -1 return -1
@@ -28,12 +28,29 @@ func BiSearch[ValueType NumberType, T Element[ValueType]](sElement []T, value Va
} }
} }
if matchUp == true { switch matchUp {
case 1:
if (sElement[mid].GetValue()) < value && if (sElement[mid].GetValue()) < value &&
(mid+1 < len(sElement)-1) { (mid+1 < len(sElement)-1) {
return mid + 1 return mid + 1
} }
return mid return mid
case -1:
if (sElement[mid].GetValue()) > value {
if mid - 1 < 0 {
return -1
} else {
return mid - 1
}
} else if (sElement[mid].GetValue()) < value {
if (mid+1 < len(sElement)-1) {
return mid + 1
} else {
return mid
}
} else {
return mid
}
} }
return -1 return -1