Javaコレクションにおける二分検索アルゴリズムの実装
3459 ワード
Javaコレクションにおける二分検索アルゴリズムの実装
Arrays.binarySearchは秩序配列の特定区間の二分検索を実現し、簡単だと思いますが、ソースコードを読むと、これらのライブラリを実現する優れたテクニックが見え、常に完璧さと効率を追求しています.
学ぶべき点は次のとおりです.
(1)境界検査;
(2)中位数を求める場合はx/2ではなくシフト操作を用いる.
(3)検索された要素が配列内にない場合、戻り値によって挿入すべき位置が示され、直接-1を返すのではなく、戻り値によって挿入すべき位置が示される.
public static int binarySearch(int[] a, int fromIndex, int toIndex,
int key) {
rangeCheck(a.length, fromIndex, toIndex);
return binarySearch0(a, fromIndex, toIndex, key);
}
//Like public version, but without range checks.
private static int binarySearch0(int[] a, int fromIndex, int toIndex,
int key) {
int low = fromIndex;
int high = toIndex - 1;
while (low <= high) {
int mid = (low + high) >>> 1;
int midVal = a[mid];
if (midVal < key)
low = mid + 1;
else if (midVal > key)
high = mid - 1;
else
return mid;//key found
}
return -(low + 1); //key not found.
}
同様にCollectionsにも同様の補助関数があるが,反復器を用いて特定の位置を取得する要素である.
Arrays.binarySearchは秩序配列の特定区間の二分検索を実現し、簡単だと思いますが、ソースコードを読むと、これらのライブラリを実現する優れたテクニックが見え、常に完璧さと効率を追求しています.
学ぶべき点は次のとおりです.
(1)境界検査;
(2)中位数を求める場合はx/2ではなくシフト操作を用いる.
(3)検索された要素が配列内にない場合、戻り値によって挿入すべき位置が示され、直接-1を返すのではなく、戻り値によって挿入すべき位置が示される.
public static int binarySearch(int[] a, int fromIndex, int toIndex,
int key) {
rangeCheck(a.length, fromIndex, toIndex);
return binarySearch0(a, fromIndex, toIndex, key);
}
//Like public version, but without range checks.
private static int binarySearch0(int[] a, int fromIndex, int toIndex,
int key) {
int low = fromIndex;
int high = toIndex - 1;
while (low <= high) {
int mid = (low + high) >>> 1;
int midVal = a[mid];
if (midVal < key)
low = mid + 1;
else if (midVal > key)
high = mid - 1;
else
return mid;//key found
}
return -(low + 1); //key not found.
}
同様にCollectionsにも同様の補助関数があるが,反復器を用いて特定の位置を取得する要素である.
public static
int binarySearch(List extends Comparable super T>> list, T key) {
if (list instanceof RandomAccess || list.size()< BINARYSEARCH_THRESHOLD)
return Collections. indexedBinarySearch(list, key);
else
return Collections. iteratorBinarySearch(list, key);
}
private static
int indexedBinarySearch(List extends Comparable super T>> list, T key ) {
int low = 0;
int high = list.size()-1;
while (low <= high) {
int mid = ( low + high) >>> 1;
Comparable super T> midVal = list.get( mid);
int cmp = midVal.compareTo( key);
if ( cmp < 0)
low = mid + 1;
else if ( cmp > 0)
high = mid - 1;
else
return mid; // key found
}
return -(low + 1); // key not found
}
private static
int iteratorBinarySearch(List extends Comparable super T>> list, T key )
{
int low = 0;
int high = list.size()-1;
ListIterator extends Comparable super T>> i = list.listIterator();
while (low <= high) {
int mid = ( low + high) >>> 1;
Comparable super T> midVal = get(i , mid );
int cmp = midVal.compareTo( key);
if ( cmp < 0)
low = mid + 1;
else if ( cmp > 0)
high = mid - 1;
else
return mid; // key found
}
return -(low + 1); // key not found
}
/**
* Gets the ith element from the given list by repositioning the specified
* list listIterator.
*/
private static T get(ListIterator extends T> i, int index) {
T obj = null;
int pos = i.nextIndex();
if (pos <= index) {
do {
obj = i.next();//
} while ( pos++ < index);
} else {
do {
obj = i.previous();
} while (-- pos > index);
}
return obj;
}