LintCode 1690. 朋友推荐(二分插入)

1. 题目

某交友网站会给除了第一个用户以外的每个新注册的用户推荐一位之前已经注册过并且性格值和他最相近的用户,如果有多人满足条件则选择性格值较小的。

给定数组val[]表示按时间顺序注册的 n 位用户的性格值,输出一个大小为 n-1 的数组,表示系统给这些人推荐的用户的性格值。

样例 1:
输入: val=[8,9,7,3,0,5,11]
输出: [8,8,7,3,3,9]
解释:
令 ans = []2 个数为 9,前面只有第 1 个数 8,此时 ans = [8]3 个数为 7,前面的数有 8, 9,与 7 性格值最小的为 8,此时 ans = [8, 8]4 个数为 3,前面的数有 8, 9, 7,与  3 性格值最小的为 7,此时 ans = [8, 8, 7]5 个数为 0,前面的数有 8, 9, 7, 3,与  0 性格值最小的为 3,此时 ans = [8, 8, 7, 3]6 个数为 5,前面的数有 8, 9, 7, 3, 0,与  5 性格值最小的为 3,此时 ans = [8, 8, 7, 3, 3]7 个数为 11,前面的数有 8, 9, 7, 3, 0, 5,与  11 性格值最小的为  9,此时 ans = [8, 8, 7, 3, 3, 9]

样例 2:
输入: val=[465, 5464, 6467, 6466779, 6461, 56]
输出: [465,5464,6467,6467,465]
解释:
令 ans = []2 个数为 5464,前面只有第 1 个数 465,此时 ans = [465]3 个数为 6467,前面的数有 465, 5464,与 6467 性格值最小的为 5464,此时 ans = [465, 5464]4 个数为 6466779,前面的数有 465, 5464, 6467,与  6466779 性格值最小的为 6467,此时 ans = [465, 5464, 6467]5 个数为 6461,前面的数有 465, 5464, 6467, 6466779,与  6461 性格值最小的为 6467,此时 ans = [465, 5464, 6467, 6467]6 个数为 56,前面的数有 465, 5464, 6467, 6466779, 6461,与  56 性格值最小的为 465,此时 ans =[465, 5464, 6467, 6467, 465]

注意事项
2<=n<=100000
0<=val<=1000000

类似题目:LeetCode 315. 计算右侧小于当前元素的个数(二叉查找树&二分查找&归并排序逆序数总结)

2. 解题

  • 给一个空数组,依次把性格值二分插入到其中
  • 检查插入位置前后跟我 绝对值较小 的取为答案
  • 变形版 二分查找请参考
class Solution {
	int l,r,mid;
public:
    vector<int> getAns(vector<int> &val) {
        if(val.size() <= 1)
        	return {};
        vector<int> t;//二分插入数组
        t.push_back(val[0]);
        vector<int> ans(val.size()-1);
        int i, idx, k = 0, f, b;
        for(i = 1; i < val.size(); i++,k++)
        {
        	idx = bs(t,val[i]);//插入位置
    		t.insert(t.begin()+idx+1,val[i]);//插入数组
    		f = (idx >= 0 ? t[idx] : -10000000);//前面的性格值
    		b = (idx+2 < t.size() ? t[idx+2] : -10000000);//后面的性格值
    		if(abs(f-val[i]) <= abs(b-val[i]))
    			ans[k] = t[idx];//取较小的
    		else
    			ans[k] = t[idx+2];
        }
        return ans;
    }

    int bs(vector<int> & a, int& target)
    {	//二分查找,性格值小于等我的 最后一个
    	l = 0, r = a.size()-1;
    	while(l <= r)
    	{
    		mid = l+((r-l)>>1);
    		if(a[mid] > target)
    			r = mid-1;
    		else //(a[mid] <= target)
    		{
    			if(mid==a.size()-1 || a[mid+1] > target)
    				return mid;
    			else
    				l = mid+1;
    		}
    	}
    	return -1;
    }
};

100% 数据通过测试
总耗时 653 ms
您的提交打败了 35.48% 的提交!

发布了724 篇原创文章 · 获赞 737 · 访问量 22万+

猜你喜欢

转载自blog.csdn.net/qq_21201267/article/details/104838218