解题思路: 这是一道很典型的贪心题目,只要每次吃的都是最先要腐烂的苹果,那么一定能够保证吃到最多的苹果,而且当前只知道目前所拥有的苹果的腐烂时间,之后要收集的苹果和其腐烂时间是不知道的。定义一个优先队列,以腐烂日期最新为排序条件,遍历每一天,先判断是否有苹果可放入,接着丢弃掉腐烂的苹果,最后再吃苹果,由于过了n天还有苹果能吃,所以要符合判断条件,遍历到n天后还有苹果能吃就继续往下走,代码如下:
class Solution {
public:
int eatenApples(vector<int>& apples, vector<int>& days) {
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> q;
int count = 0, i = 0;
int n = apples.size();
while(i < apples.size() || !q.empty()) {
if(i < n && apples[i] != 0 && days[i] != 0) {
q.push({i + days[i], apples[i]});
}
while(!q.empty() && q.top().first <= i) {
q.pop();
}
if(!q.empty()) {
count ++;
auto [a, b] = q.top();
q.pop();
if(b > 1) {
q.push({a, b - 1});
}
}
i ++;
}
return count;
}
};
|