算法刷題手記3

描述

二分查找擴展:如果數組有序,且存在多個目標值,查詢所有目標值索引返回。

public class Main {

    public static void main(String[] args) {
        ArrayList list = new ArrayList();
        int[] arr = {1, 2, 2, 2, 3, 4, 5, 6, 7};
        List list1 = binarySearch(arr, 0, arr.length, 2, list);
        System.out.println(Arrays.toString(list1.toArray()));

    }

    public static List<Integer> binarySearch(int[] arr, int left, int right, int findVal, List<Integer> list) {
        int mid = (left + right) / 2;
        int midVal = arr[mid];

        if (findVal < midVal) {
            right = mid - 1;
            return binarySearch(arr, left, right, findVal, list);
        } else if (findVal > midVal) {
            left = mid + 1;
            return binarySearch(arr, left, right, findVal, list);
        } else {
            left = mid - 1;
            while (left >= 0) {
                if (arr[left] != findVal) {
                    break;
                }
                list.add(left);
                left--;
            }
            list.add(mid);
            right = mid + 1;
            while (right < arr.length) {
                if (arr[right] != findVal) {
                    break;
                }
                list.add(right);
                right++;
            }
            return list;
        }
    }
    
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章