lintcode 简单查找

最新推荐文章于 2020-06-14 22:26:13 发布

锤某

最新推荐文章于 2020-06-14 22:26:13 发布

阅读量233

点赞数

本文链接：https://blog.youkuaiyun.com/qq_40147449/article/details/93376861

版权

描述

对于两个数组，第一个数组可能包含可重复的整数，第二个数组是第一个数组的子集。返回的数组的长度与第二个数组相同。对于第二个数组中的每个元素a，第一个数组中有多少个数字<=a。

样例

1

输入：nums = [3, 2, 4, 3, 5, 1]，sub = [2, 4]
输出：[2, 5]
解释：<=2的数字[1,2]，<=4的数字[1,2,3,3,4]

2

输入：nums = [3, 1, 2, 3, 3, 1]，sub = [1,3]
输出：[2,6]
解释：<=1的数字[1,1]，<=3的数字[1,1,2,3,3,3]

思路

两种方法

1

暴力的遍历两个数组

2

将第一个数组排序，然后使用upper_bound函数计算小于等于sub数组中的数

代码

1

class Solution {
public:
    /**
     * @param nums:
     * @param sub:
     * @return: return a Integer array
     */
    vector<int> SimpleQueries (vector<int> &nums, vector<int> &sub) {
        // write your code here


        vector<int> res;
        for (int i = 0; i < sub.size(); i++){
            int temp = sub[i];
            int n = 0;
            for (int j = 0; j < nums.size(); j++) {
                if (temp >= nums[j])
                    n++;
            }
            res.push_back(n);
        }
        return res;
    }
};

2

class Solution
{
public:
    /**
     * @param nums:
     * @param sub:
     * @return: return a Integer array
     */
    vector<int> SimpleQueries (vector<int> &nums, vector<int> &sub)
    {
        sort(nums.begin(), nums.end());
        vector<int> res;
        for(int i; i < sub.size(); i++) {
            int s = sub[i];
            res.push_back(upper_bound(nums.begin(), nums.end(), s) - nums.begin());
        }
        return res;
    }
};