二分搜索常用变种(N个相同目标,寻找第一个或最后一个目标元素)leetcode 278

 

    /*
     * 二分搜索的一个变种,找到第一个等于target的下标
     */
    int binarySearchFirst(vector& nums, int target) {
        if (nums.empty()) {
            return -1;
        }
        
        int left = 0;
        int right = nums.size() - 1;
        while (left < right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] < target) {
                left = mid + 1;
            }
            else {
                right = mid;
            }
        }
        
        if (nums[left] == target) {
            return left;
        } 
        else {
            return -1;
        }
    }
    
    /*
     * 二分搜索的一个变种,找到第一个等于target的下标
     */

    int binarySearchLast(vector& nums, int target) {
        if (nums.empty()) {
            return -1;
        }
        
        int left = 0;
        int right = nums.size() - 1;
        while (left < right) {
            int mid = left + (right + 1 - left) / 2;
            if (nums[mid] > target) {
                right = mid - 1;
            }
            else {
                left = mid;
            }
        }
        
        if (nums[right] == target) {
            return right;
        }
        else {
            return -1;
        }
    }
    
    vector searchRange(vector& nums, int target) {
        vector result;
        int first = binarySearchFirst(nums, target);
        int last = binarySearchLast(nums, target);
        result.push_back(first);
        result.push_back(last);
        return result;
    }

 

你可能感兴趣的:(算法策略,C/C++,leetcode,linux)