【牛客网-名企高频面试题】 NC105 二分查找

NC105 二分查找

题目描述

请实现有重复数字的有序数组的二分查找。
输出在数组中第一个大于等于查找值的位置，如果数组中不存在这样的数，则输出数组长度加一。

示例1
输入

5,4,[1,2,4,4,5]

返回值

3

说明

输出位置从1开始计算

import java.util.*;


public class Solution {
    /**
     * 二分查找
     * @param n int整型 数组长度
     * @param v int整型 查找值
     * @param a int整型一维数组 有序数组
     * @return int整型
     */
    public int upper_bound_ (int n, int v, int[] a) {
        // write code here
        int i = 0;
        int j = n - 1;
        int mid = i + (j - 1)/ 2;
        while(i <= j){
            if(a[mid] < v){
                i = mid + 1;
                mid = (i + j)/2;
            }else if(mid > 0 && a[mid - 1] >= v){
                j = mid - 1;
                mid = (i + j)/2;
            }else{
                return mid + 1;
            }
            
        }
        return n + 1;
    }
}