前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >天池 在线编程 最佳利用率(二分查找 + 哈希)

天池 在线编程 最佳利用率(二分查找 + 哈希)

作者头像
Michael阿明
发布2021-09-06 10:12:26
2320
发布2021-09-06 10:12:26
举报
文章被收录于专栏:Michael阿明学习之路

1. 题目

给两个排序的数组。 从两个数组中各取取一个数,这两个数之和需要小于或等于k, 需要找到两数之和最大的索引组合。 返回一对包含两个列表的索引。 如果有多个两数之和相等的索引答案,你应该选择第一个数组索引最小的索引对。 在此前提下,你应该选择第二个数组索引最小的索引对。

1)两数的总和<= k 2)总和是最大的 3)两个数组索引都尽量最小

如果你无法找到答案,你应返回一个空列表[]。 You can assume that the numbers in arrays are all positive integer or zero.

代码语言:javascript
复制
示例1
输入:
A = [1, 4, 6, 9], B = [1, 2, 3, 4], K = 9
输出: 
[2, 2]

示例2:
输入: 
A = [1, 4, 6, 8], B = [1, 2, 3, 5], K = 12
输出:
[2, 3]

2. 解题

  • 遍历 B 的元素,记录每个值第一次出现的位置
  • 遍历 A,在 B 中查找小于等于 K-A[i] 的最大的数
  • 时间复杂度 O(n1​logn2​)
代码语言:javascript
复制
class Solution {
public:
    /**
     * @param A: an integer sorted array
     * @param B: an integer sorted array
     * @param K: an integer
     * @return: return a pair of index
     */
    vector<int> optimalUtilization(vector<int> &A, vector<int> &B, int K) {
        // write your code here
        if(A.empty() || B.empty() || A[0]+B[0] > K) return {};
        unordered_map<int,int> m;
        for(int i = 0; i < B.size(); ++i)
            if(m.find(B[i]) == m.end())
                m[B[i]] = i;//每个数最早出现的位置
        int i = 0, j = B.size()-1, n1 = A.size(), n2 = B.size(), sum, maxsum=INT_MIN;
        vector<int> ans(2);
        for(int i = 0; i < n1; ++i)
        {
            int t = K-A[i];
            int l = 0, r = n2-1, pos=-1;
            while(l <= r)	// 二分查找
            {
                int mid = l+((r-l)>>1);
                if(B[mid] > t)
                    r = mid-1;
                else
                {
                    if(mid==n2-1 || B[mid+1] > t)
                    {
                        pos = mid;
                        break;
                    }
                    else
                        l = mid+1;
                }
            }
            if(pos != -1)
            {
                sum = A[i]+B[pos];
                if(sum > maxsum)
                {
                    maxsum = sum;
                    ans[0] = i;
                    ans[1] = m[B[pos]];
                }
            }
            else
                break;
        }
        return ans;
    }
};

50ms C++

我的CSDN博客地址 https://michael.blog.csdn.net/

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021/03/13 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档