吃苹果的最大数目-c语言贪心算法

吃苹果的最大数目-c语言贪心算法

有一棵特殊的苹果树，一连 n 天，每天都可以长出若干个苹果。在第 i 天，树上会长出 apples[i] 个苹果，这些苹果将会在 days[i] 天后（也就是说，第 i + days[i] 天时）腐烂，变得无法食用。也可能有那么几天，树上不会长出新的苹果，此时用 apples[i] == 0 且 days[i] == 0 表示。

你打算每天 最多 吃一个苹果来保证营养均衡。注意，你可以在这 n 天之后继续吃苹果。

给你两个长度为 n 的整数数组 days 和 apples ，返回你可以吃掉的苹果的最大数目。

示例 1：

输入：apples = [1,2,3,5,2], days = [3,2,1,4,2]
输出：7
解释：你可以吃掉 7 个苹果：

• 第一天，你吃掉第一天长出来的苹果。
• 第二天，你吃掉一个第二天长出来的苹果。
• 第三天，你吃掉一个第二天长出来的苹果。过了这一天，第三天长出来的苹果就已经腐烂了。
• 第四天到第七天，你吃的都是第四天长出来的苹果。

示例 2：

输入：apples = [3,0,0,0,0,2], days = [3,0,0,0,0,2]
输出：5
解释：你可以吃掉 5 个苹果：

• 第一天到第三天，你吃的都是第一天长出来的苹果。
• 第四天和第五天不吃苹果。
• 第六天和第七天，你吃的都是第六天长出来的苹果。

typedef struct{
    int apples;
    int days;
}Heap;

Heap** heap;
int heapSize;

void swap(int a, int  b){
    Heap* tmp = heap[a];
    heap[a] = heap[b];
    heap[b] = tmp;
}

void Adjustdown_heap(){
    int parents = 1;        //下滤。 parents为1
    int child = 2*parents;
    while ( child < heapSize ){
        if ( child+1<heapSize && heap[child+1]->days < heap[child]->days){
            child++;
        }
        if ( heap[child]->days > heap[parents]->days ) break;
        swap(child,parents);
        parents = child;
        child = 2*parents;
    }
}

void Adjustup_heap(){
    int child = heapSize-1;
    int parents = child/2;
    while ( parents > 0 ){
        if ( heap[child]->days > heap[parents]->days ) break;
        swap(child,parents);
        child = parents;
        parents = child/2;
    }

}

void pop_heap(){
    heap[1] = heap[heapSize-1];
    heapSize--;
    Adjustdown_heap();
}

void add_heap(int apples, int days){
    Heap* new = malloc(sizeof(Heap));
    new->apples = apples;
    new->days = days;
    heap[heapSize++] = new;
    Adjustup_heap();
}

bool empty_heap(){
    return heapSize <= 1;
}

Heap* getTop_heap(){
    return heap[1];
}


int eatenApples(int* apples, int applesSize, int* days, int daysSize){
    //用产生的苹果和天数的取小值来表示未来的n天中你都有苹果可以吃。
    //维护这个n值就可以了，最后再把剩余的n给加上去
    //上述思路有个bug:那就是吃的苹果是按照顺序过来的。而不考虑腐烂时间。所以导致后来的苹果快腐烂而没吃到

    //所以这个题目就必须得用贪心思想了。首先要改变一下day数组。将其含义表示成在第i天开始腐烂。
    int ans = 0;
    int i = 0;
    for(int k = 0; k<daysSize; k++){
        days[k] += k;
    }

    heap = malloc(sizeof(Heap*)*(applesSize+1));
    heapSize = 1;

    //模拟过程.每天加入一批苹果.过了腐烂天数的就置零.没有腐烂的就从快要腐烂的中挑出一个苹果吃掉.
    while( i < daysSize || !empty_heap() ){         //天数未到或者堆不空.
        if ( i < daysSize ) add_heap(apples[i],days[i]);       //加入新的一批苹果.
        while ( !empty_heap() && ( getTop_heap()->days <= i || getTop_heap()->apples <= 0) ){
            pop_heap();                               //清理堆中腐烂的水果和没有水果的批次
        }
        i++;
        if ( empty_heap() ) continue;
        (getTop_heap()->apples)--;
        ans++;
    }   
    return ans;
}