力扣(LeetCode)2530. 执行 K 次操作后的最大分数(C++)
贪心+优先队列
请看答案需求:得到最大分数。易猜到,得到最大分数的取法是每次取数组中最大的数字(贪心思路)。
问题转化为:如何快速找到数组中最大的数字,根据问题规模 k = 1 0 5 k=10^5 k=105,维护优先队列即可 O ( k l o g 2 n ) O(klog_2n) O(klog2n)解决问题。
请看如下代码:
class Solution {
public:long long maxKelements(vector<int>& nums, int k) {// priority_queue<int> pq(nums.begin(), nums.end());priority_queue<int> pq(less<int>(), move(nums));long long ans = 0;while (k --) {int t = pq.top();pq.pop();ans += t;t = (t + 2) / 3;pq.push(t);}return ans;}
};
时间复杂度 O ( n + k l o g n ) O(n+klogn) O(n+klogn):维护优先队列,的时间复杂度 O ( n ) O(n) O(n)。
空间复杂度 O ( 1 ) O(1) O(1):只使用常数级空间。
致语
- 理解思路很重要。
- 请读者放心留言,可以是疑惑的点,或者讨论!!墨染看到会回复的。