https://leetcode.com/problems/maximum-subsequence-score/description/
2542. Maximum Subsequence Score
給你兩個大小一樣的陣列 nums1 和 nums2,以及一個數字 k,找出大小為 k 的子序列,
透過下列規則計算出分數 score:
score = (nums1[i0] + nums1[i1] +...+ nums1[ik - 1]) *
min(nums2[i0] , nums2[i1], ... ,nums2[ik - 1])
求出 score 最大是多少。
Example 1:
Input: nums1 = [1,3,3,2], nums2 = [2,1,3,4], k = 3
Output: 12
Explanation:
The four possible subsequence scores are:
- We choose the indices 0, 1, and 2 with score = (1+3+3) * min(2,1,3) = 7.
- We choose the indices 0, 1, and 3 with score = (1+3+2) * min(2,1,4) = 6.
- We choose the indices 0, 2, and 3 with score = (1+3+2) * min(2,3,4) = 12.
- We choose the indices 1, 2, and 3 with score = (3+3+2) * min(1,3,4) = 8.
Therefore, we return the max score, which is 12.
思路:
1.最大分數需要 nums1 的和越高越好,nums2 的最小數越大越好,兩者相乘才會最高,簡
單講就是要求長度為k的最優Pair。
2.我們可以先讓Pair的其中一邊是最佳解,再嘗試把其他nums1的元素加進來比較,如果得
到更大分數,就把比較小的鍵值對移除得到更優解。
3.因為分數是 nums2 的「其中一個數字」乘以nums1的「多個數字之和」,所以nums2變化
比較小,我們把nums2由大到小排序並取k個Pair可以得到一個局部最佳解ans。
4.從k+1位置開始,每次嘗試加入一個新Pair,如果加入後的和變更大就把序列裡的最小
值移除加入新的Pair,並計算nums2次優和nums1更優解相乘是否更大,更新ans。
5.把nums1的序列放進一個MinHeap,這樣如果和更大,每次pop出來的都是最小元素。
Java Code: