插值查找算法

插值查找算法原理

• 插值查找算法类似于二分查找,不同的是插值查找每次从自适应mid处开始查找
• 将折半查找中的求mid索引的公式 (mid=frac{left+right}{2}=low+frac{1}{2}(left+right)) 改成 (mid=low+frac{findVal-arr[left]}{arr[right]-arr[left]}(right-left))

举例说明插值查找算法

代码实现

代码实现返回查找到的第一个索引

public class InsertValSearch {
    public static void main(String[] args) {
        int[] arr = {1, 8, 10, 89, 1000, 1234};
        int index = insertValSearch(arr, 0, arr.length - 1, 1000);
        System.out.println("index = "+index);
    }
    /**
     * @param arr     数组
     * @param left    左边的下标
     * @param right   右边的下标
     * @param findVal 待查找的值
     * @return 返回找到的第一个索引
     */
    private static int insertValSearch(int[] arr, int left, int right, int findVal) {
        if (findVal < arr[left] || findVal > arr[right] || left > right) {
            return -1;
        }
        int mid = left + (findVal - arr[left]) / (arr[right] - arr[left]) * (right - left);
        if (findVal < arr[mid]) {
            return insertValSearch(arr, left, mid - 1, findVal);
        } else if (findVal > arr[mid]) {
            return insertValSearch(arr, mid + 1, right, findVal);
        }else {
            return mid;
        }
    }
}

返回查找到的所有下标

public class InsertValSearch {
    public static void main(String[] args) {
        int[] arr = {1, 8, 10, 89, 1000,1000, 1234};
        List<Integer> indexList = insertValSearch2(arr, 0, arr.length - 1, 1000);
        System.out.println("indexList = "+indexList);
    }

    /**
     * @param arr     数组
     * @param left    左边的下标
     * @param right   右边的下标
     * @param findVal 待查找的值
     * @return 返回找到的所有值的下标
     */
    private static List<Integer> insertValSearch2(int[] arr, int left, int right, int findVal) {
        if (findVal < arr[left] || findVal > arr[right] || left > right) {
            return new ArrayList<Integer>();
        }
        int mid = left + (findVal - arr[left]) / (arr[right] - arr[left]) * (right - left);
        List<Integer> indexList = new ArrayList<>();
        if (findVal < arr[mid]) {
            return insertValSearch2(arr, left, mid - 1, findVal);
        } else if (findVal > arr[mid]) {
            return insertValSearch2(arr, mid + 1, right, findVal);
        } else {
            // 这里需要返回所有的下标,所以需要对找到的一个下标进行向左和向右的遍历
            int temp = mid - 1;
            while (true) { // 向左遍历
                if (temp < 0 || arr[temp] != findVal) {
                    break;
                }
                indexList.add(temp);
                temp -= 1;
            }
            indexList.add(mid);
            temp = mid + 1;
            while (true) { //向右遍历
                if (temp > arr.length - 1 || arr[temp] != findVal) {
                    break;
                }
                indexList.add(temp);
                temp += 1;
            }

            return indexList;
        }
    }

插值查找算法注意事项

• 对于数据量较大,关键字分布比较均匀的查找表来说,采用插值查找,速度较快
• 关键字分布不均匀的情况下,该方法不一定比折半查找好