qbit/go
1
0
Fork 0
mirror of https://github.com/golang/go synced 2024-11-21 23:24:41 -07:00
Code Releases Activity

index/suffixarray: use sort.Search

Browse Source 
R=gri
CC=golang-dev
https://golang.org/cl/3200041
This commit is contained in:
Russ Cox 2010-11-19 16:04:25 -05:00
parent 07791d04d6
commit b51f0c5cca
1 changed files with 2 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/pkg/index/suffixarray/suffixarray.go
View File

@ -54,21 +54,8 @@ func (x *Index) at(i int) []byte {
}
// Binary search according to "A Method of Programming", E.W. Dijkstra.
func (x *Index) search(s []byte) int {
i, j := 0, len(x.sa)
// i < j for non-empty x
for i+1 < j {
// 0 <= i < j <= len(x.sa) && (x.at(i) <= s < x.at(j) || (s is not in x))
h := i + (j-i)/2 // i < h < j
if bytes.Compare(x.at(h), s) <= 0 {
i = h
} else { // s < x.at(h)
j = h
}
}
// i+1 == j for non-empty x
return i
return sort.Search(len(x.sa), func(i int) bool { return bytes.Compare(x.at(i), s) >= 0 })
}
@ -84,12 +71,7 @@ func (x *Index) Lookup(s []byte, n int) []int {
if len(s) > 0 && n != 0 {
// find matching suffix index i
i := x.search(s)
// x.at(i) <= s < x.at(i+1)
// ignore the first suffix if it is < s
if i < len(x.sa) && bytes.Compare(x.at(i), s) < 0 {
i++
}
// x.at(i-1) < s <= x.at(i)
// collect the following suffixes with matching prefixes
for (n < 0 || len(res) < n) && i < len(x.sa) && bytes.HasPrefix(x.at(i), s) {