代码题(1)— 两数之和、lower_bound和upper_bound算法

 1、两数之和

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target  的那 两个 整数,并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。你可以按任意顺序返回答案。

输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。

输入:nums = [3,2,4], target = 6
输出:[1,2]
class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> res;
        if(nums.empty())
            return res;
        unordered_map<int, int> nums_map;
        for(int i=0;i<nums.size();++i){
            nums_map[nums[i]] = i;
        }
        for(int i=0;i<nums.size();++i){
            int diff = target - nums[i];
            if(nums_map.count(diff) && nums_map[diff] != i){
                res.push_back(i);
                res.push_back(nums_map[diff]);
                break;
            }
        }
        return res;
    }
};

简化版本

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> res;
        if(nums.empty())    
            return res;
        unordered_map<int, int> nums_map;
        for(int i=0;i<nums.size();++i){
            auto it = nums_map.find(target - nums[i]);
            if(it != nums_map.end()){
                res.push_back(i);
                res.push_back(it->second);
                return res;
            }
            nums_map[nums[i]] = i;
        }
        return res;
    }
};

2、lower_bound:查找序列中的第一个出现的值大于等于val的位置

  这个序列中可能会有很多重复的元素,也可能所有的元素都相同,为了充分考虑这种边界条件,STL中的lower_bound算法总体上是才用了二分查找的方法,但是由于是查找序列中的第一个出现的值大于等于val的位置,所以算法要在二分查找的基础上做一些细微的改动。

     首先是我修改数据结构课本上的二分查找实现的lower_bound算法:

int lower_bound(vector<int> array, int size, int key)
{
    int low = 0, high = size - 1;
    int mid, pos = 0;  //需要pos记录位置
    while (low < high)
    {
        mid = (low + high) / 2;
        if (array[mid] < key)  //若中位数的值小于key的值,我们要在右边子序列中查找,这时候pos可能是右边子序列的第一个
        {
            low = mid + 1;
            pos = low;
        }
        else
        {
            high = mid;  //若中位数的值大于等于key,我们要在左边子序列查找,但有可能middle处就是最终位置,所以我们不移动last
            pos = high;
        }
    }
    return pos;
}

3、upper_bound:返回的是最后一个大于等于val的位置,也是有一个新元素val进来时的插入位置。

int upper_bound(vector<int> array, int size, int key)
{
    int low = 0, high = size - 1;
    int mid, pos = 0;
    while (low < high)
    {
        mid = (low + high) / 2;
        if (array[mid] <= key)
        {
            low = mid + 1;
            pos = low;
        }
        else
        {
            high = mid;
            pos = high;
        }
    }
    return pos;
}

 

posted @ 2018-06-01 18:16  深度机器学习  阅读(359)  评论(0编辑  收藏  举报