poj_1042 贪心算法

poj 1042 gone fishing

题目要求:

   由有n个湖, 按照顺序排列,一个人从第一个湖向最后一个湖行进(方向只能从湖0到湖n-1),途中可以在湖中钓鱼。在每个湖中钓鱼时,开始的5分钟内可以钓到 f[i] 条,之后每5分钟钓到的鱼数目递减 d[i] ,且每两个相邻的湖的距离 t[i] 给出(t[i] 表示 由第 i 个湖向第 i + 1个湖行进在路上花费5分钟的倍数的时间, 即如果t[3] = 4,则表示从第三个湖向第四个湖行进需要花费20分钟)。 现给定总时间h 小时,求出在每个湖钓鱼时间的最好的方案,使得钓鱼的总数最多,如果两种方案结果相同,则输出在较小序号的湖中钓鱼时间更多的那个。

题目分析:

  最优化问题,可以选择动态规划或者贪心算法,初看起来,是个多阶段的决策问题,且每个阶段会对下一个阶段产生影响,并不好找到合适的贪心策略,更适合采用动态规划。先考虑动态规划算法,可以考虑 动规数组 dp[t][n] 表示在 t 时刻到达湖 n 时之前钓鱼结果的最优值,可以找出递推关系式 dp[t][n] =  max{ dp[t1][n-1] + fish_result(n-1, t - t1 - t[n-1])}

即对到达湖 n - 1的时刻 t1 进行枚举(在一定的范围之内,小于 t - t[n-1], 大于xx),然后取出最大值作为 dp[t][n],这样从 n - 1 推到 n。

  个人认为这种动态规划的方法是可行的,但是。。由于水平太菜。。没有成功。。

  考虑采用贪心的算法来实现,仔细分析一下,当确定了需要到达的湖的个数n,也就确定了在路上需要消耗的时间,从而得到钓鱼的总时间,这样在规定的时间内钓鱼,每次都选择当前5分钟内收益最大的那个湖进行钓鱼(可以采用优先级队列  priority_queue 来实现)。 考虑到这样做会有可能出现先在湖 A 中钓鱼,再在湖B中钓鱼,然后再回到湖A中钓鱼... 但是这样不会影响最终的结果(这也是贪心算法可以运用到此题的前提)。因为,虽然只能从湖0到湖n-1行进,此时进行贪心选择虽然每次选择的顺序可能不断颠倒变化,但是这只是确定湖A有几次被选中钓鱼,也就是确定从湖0到湖n-1每个湖中钓鱼的时间。在实际行进途中钓鱼的时候,就按照贪心算出来的方案(在湖 i 中钓鱼几次),在途中的湖中钓鱼。

  通过固定要经过的湖的个数n,将每个湖由于在路上消耗时间的差异去掉,使得面临的选择无差异,可以运用贪心算法。

实现代码:

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<queue>
#include<vector>
using namespace std;
#define MAX_INTERVAL_NUM 200
#define MAX_LAKE_NUM 25

int dist_time[MAX_LAKE_NUM];
int min_dist_time[MAX_LAKE_NUM];
int stay_times[MAX_LAKE_NUM];
int init_num[MAX_LAKE_NUM];
int dec_num[MAX_LAKE_NUM];

struct LakeNode{
    int lake_index;
    int cur_fish_amount;
    int dec_amount;
    int stay_time;
};
LakeNode lake_nodes[MAX_LAKE_NUM];
struct cmp{
    bool operator() (LakeNode* lake1, LakeNode* lake2){
        if (lake1->cur_fish_amount == lake2->cur_fish_amount){
            return lake1->lake_index > lake2->lake_index;
        }
        return lake1->cur_fish_amount < lake2->cur_fish_amount;
    }
};

void ClearQueue(priority_queue<LakeNode*, vector<LakeNode*>, cmp>& queue){
    while(! queue.empty()){
        queue.pop();
    }
}


int Resolve(int N, int T){
    priority_queue<LakeNode*, vector<LakeNode*>, cmp> lake_queue;
    LakeNode* lake;
    int result = 0, max_amount = 0;
    stay_times[0] = T;
    for (int n = 1; n <= N; n ++){
        result = 0;
        int fish_time = T - min_dist_time[n - 1];
        if (fish_time <= 0){
            break;
        }
        ClearQueue(lake_queue);
        for(int i = 0; i < n; i ++){
            lake_nodes[i].stay_time = 0;
            lake_nodes[i].lake_index = i;
            lake_nodes[i].cur_fish_amount = init_num[i];
            lake_nodes[i].dec_amount = dec_num[i];

            lake_queue.push((LakeNode*)&lake_nodes[i]);
        }
        while(! lake_queue.empty() && fish_time > 0){
            lake = lake_queue.top();
            lake_queue.pop();

            if (lake->cur_fish_amount > 0){
                result += lake->cur_fish_amount;
                lake->cur_fish_amount -= lake->dec_amount;
            }
            if (lake->cur_fish_amount <= 0){
                lake->cur_fish_amount = 0;
            }

            lake->stay_time ++;
            fish_time --;
            lake_queue.push(lake);

        }
        if (result > max_amount){
            max_amount = result;
            for(int i = 0; i < n; i ++){
                stay_times[i] = lake_nodes[i].stay_time;
            }
        }
    }
    return max_amount;
}

int main(){
    int n, h;

    while(true){
        cin >> n;
        if (n == 0){
            break;
        }
        cin >> h;
        h = h*12;
        for(int i = 0; i < n ; i ++){
            cin >> init_num[i];
            stay_times[i] = 0;
        }
        for(int i = 0; i < n; i ++){
            cin >> dec_num[i];
        }
        min_dist_time[0] = 0;
        cin >> dist_time[0];
        for(int i = 1; i < n - 1; i ++){
            cin >> dist_time[i];
            min_dist_time[i] = min_dist_time[i - 1] + dist_time[i-1];
        }
        min_dist_time[n-1] = min_dist_time[n-2] + dist_time[n-2];

        int result = Resolve(n, h);
        cout << stay_times[0]*5 ;
        for(int i = 1; i < n; i ++){
            cout << ", " << stay_times[i]*5;
        }
        cout << endl << "Number of fish expected: " << result << endl << endl;
    }
    return 0;
}
原文地址:https://www.cnblogs.com/gtarcoder/p/4555939.html