來練習一下C++ 不然真的要生鏽了
剛好今天題目比較簡單 用maxheap裝題目測資
之後一直pop一直加就好了 時間複雜度O(N)
class Solution {
public:
long long maxKelements(vector<int>& nums, int k) {
long long int sum = 0;
priority_queue<long long int> n;
// O(n)
for (const int& i : nums)
n.push(i);
// O(k logn)
while (k) {
double temp = n.top();
sum += temp;
n.pop();
n.push(ceil(temp / 3));
k -= 1;
}
return sum;
}
};
一二三四五