每日一题-力扣（leetcode）2542. 最大子序列的分数

题目传送门

题目描述

给你两个下标从 0 开始的整数数组 nums1 和 nums2 ，两者长度都是 n ，再给你一个正整数 k 。你必须从 nums1 中选一个长度为 k 的 子序列 对应的下标。

对于选择的下标 i0 ，i1 ，…， ik - 1 ，你的 分数 定义如下：

nums1 中下标对应元素求和，乘以 nums2 中下标对应元素的 最小值 。
用公示表示： (nums1[i0] + nums1[i1] +…+ nums1[ik - 1]) * min(nums2[i0] , nums2[i1], … ,nums2[ik - 1]) 。
请你返回 最大 可能的分数。

一个数组的 子序列 下标是集合 {0, 1, …, n-1} 中删除若干元素得到的剩余集合，也可以不删除任何元素。

解题思路

自定义排序加小根堆
首先建立一个索引数组，将索引数组按照 nums2 数组从大到小的顺序自定义排序，然后遍历排好序的索引数组，将nums2 按照从大到小的顺序枚举可能的最小值，这样只需要从该最小值所在下标左侧选 k - 1 个 nums1 元素使其和最大，为了让分数尽可能地大，我们尽可能地选择 nums1[…] 中最大的 k 个数，因此用一个小根堆来维护当前能得到的最大的 k 的数。
时间复杂度为 O(nlogn)

AC代码

class Solution {
public:
    using LL = long long;
    long long maxScore(vector<int>& nums1, vector<int>& nums2, int k) {
        int n = nums1.size();
		int idxs[n]; // 建立索引数组
		for (int i = 0; i < n; i++) idxs[i] = i;
		sort(idxs, idxs + n, [&](const int &i, const int &j){
			return nums2[i] > nums2[j];
		});
        // for(auto i : idxs) cout << nums1[i] << " ";
		priority_queue<int, vector<int>, greater<int>>minHeap; // 建立小根堆
		LL sum = 0, res = 0, ans = 0;
		for(auto i : idxs) {
			if(minHeap.size() > k - 1) {
				sum -= minHeap.top(); // 去掉最小的数
				minHeap.pop();
			}
			sum += nums1[i];
			minHeap.push(nums1[i]);
			if(minHeap.size() == k) {
				res = sum * nums2[i];
				ans = max(ans, res);
			}
		}
		return ans;
    }
};