【一次过】Lintcode 462. 目标出现总和

给一个升序的数组，以及一个target，找到它在数组中出现的次数。

样例

样例1：

输入：[1,3,3,4,5]和target= 3，
输出：2。

样例2：

输入：[2,2,3,4,6]和target= 4，
输出：1。

样例3：

输入：[1,2,3,4,5]和target= 6，
输出：0。

挑战

时间复杂度在O(logn)内

---

解题思路：

明显思路是二分搜索。需要注意的是搜索到对应元素后，再向两边搜寻得到结果。

public class Solution {
    /**
     * @param A: A an integer array sorted in ascending order
     * @param target: An integer
     * @return: An integer
     */
    public int totalOccurrence(int[] A, int target) {
        // write your code here
        int l = 0, r = A.length - 1;
        if(A.length == 0 || target < A[0] || target > A[A.length - 1])
            return 0;
        
        while(l < r){
            int mid = l + (r - l)/2;
            if(target == A[mid]){
                int res = 1;
                
                int rMid = mid;
                while(rMid + 1 < A.length && A[++rMid] == target)
                    res++;
                
                int lMid = mid;
                while(lMid - 1 >= 0 && A[--lMid] == target)
                    res++;
                    
                return res;
            }else if(target > A[mid])
                l = mid + 1;
            else
                r = mid;
        }
        
        return 0;
    }
}