在排序数组中找最接近的K个数

给一个目标数 target, 一个非负整数 k, 一个按照升序排列的数组 A。在A中找与target最接近的k个整数。返回这k个数并按照与target的接近程度从小到大排序,如果接近程度相当,那么小的数排在前面。

样例

样例 1:

输入: A = [1, 2, 3], target = 2, k = 3
输出: [2, 1, 3]

样例 2:

输入: A = [1, 4, 6, 8], target = 3, k = 3
输出: [4, 1, 6]

挑战

O(logn + k) 的时间复杂度

注意事项

  1. k是一个非负整数,并且总是小于已排序数组的长度。
  2. 给定数组的长度是正整数, 不会超过 10^410​4​​
  3. 数组中元素的绝对值不会超过 10^410​4​​

输入测试数据 (每行一个参数)如何理解测试数据?

 


bool cmp(pair<int, int> a, pair<int, int> b) 
{
    if(a.second < b.second)
        return true;
    else if(a.second > b.second)
        return false;
    if(a.first < b.first)
        return true;
    return false;
	
}



class Solution {
public:
    /**
     * @param A: an integer array
     * @param target: An integer
     * @param k: An integer
     * @return: an integer array
     */
    vector<int> kClosestNumbers(vector<int> &A, int target, int k) 
    {
        // write your code here
        vector<int> ret;
        if(k == 0)
            return ret;
        unordered_map<int, int> mymap;
        int size = A.size();
        for(int i = 0; i < size; i++)
        {
            mymap[A[i]] = abs(A[i] - target);
        }
        
        vector<pair<int, int>> vc;
        for(auto mit=mymap.begin(); mit!=mymap.end(); mit++)
        {
    		vc.push_back(pair<int, int>(mit->first, mit->second));
        }
        
        sort(vc.begin(), vc.end(), cmp);
        //int count = 0;
        for(auto vit=vc.begin(); vit!=vc.end(); vit++) 
        {
		    //cout<<vit->first<<"  "<<vit->second<<endl;
		    ret.push_back(vit->first);
            if(ret.size() == k)
                break;
        }
    
        return ret;
    }
};

 

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值