USC Problem A 钓鱼 贪心

题目链接：http://61.187.179.71:9988/problem.php?cid=1104&pid=0

题目意思：有n个小湖序号从1到n，每个小湖每5分钟能钓到fi数量的鱼，但每过5分钟在此小湖钓到的鱼的数量就会减少di，从小湖i到i+1需要的时间为ti，不能回头，也就是不能从i+1号小湖到i号小湖，给你h小时，问最多能钓到多少鱼，最后钓鱼的人可能停到任意小湖。

解题思路：一开始还以为是简单的dp，写了写，发现复杂度爆炸还超级难写，后来发现是贪心题。。。
我们可以枚举钓鱼人最后停到哪个小湖，直接用总时间减去到最后一个小湖的时间，这样就不用考虑走路花费的时间了。
假设钓鱼人最后停到了小湖i，我们每次从1-i号小湖中选取可以钓到最多鱼的小湖去钓鱼，选取完之后减去相应的di，重复做这样的操作直到时间为0，得到钓鱼人最后停到小湖i时能钓到的鱼。
最后枚举i我们便能得到结果，但如果枚举求可以钓到鱼最多的小湖会超时，所以我们要用优先队列来维护。特殊：如果钓到的鱼数量一样则优先弹出序号小的。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <string>
#include <vector>
#include <list>
#include <map>
#include <queue>
#include <stack>
#include <algorithm>
#include <numeric>
#include <functional>
#define RI(N) scanf("%d",&(N))
#define RII(N,M) scanf("%d %d",&(N),&(M))
#define RIII(N,M,K) scanf("%d %d %d",&(N),&(M),&(K))
#define mem(a) memset((a),0,sizeof(a))
using namespace std;
const int inf=1e9;
const int inf1=-1*1e9;
typedef long long LL;

struct P
{
    int val;
    int poi;
};
struct cmp
{
    bool operator()(P p1,P p2)
    {
        if(p1.val==p2.val) return p1.poi>p2.poi ;
        else return p1.val<p2.val;
    }

};
priority_queue<P,vector<P>,cmp >pq;

int main()
{
    int n;
    while(RI(n)!=EOF&&n)
    {
        int h;
        RI(h);
        h=h*60;
        int f[35];
        int d[35];
        int t[35];
        for(int i=0; i<n; i++)
            RI(f[i]);
        for(int i=0; i<n; i++)
            RI(d[i]);
        for(int i=1; i<n; i++)
            RI(t[i]);
        t[0]=0;
        int ans=-1;
        int ans1[35];
        mem(ans1);
        int h1=h;
        for(int i=0; i<n; i++)
        {
            if(h1-t[i]*5<=0) break;
            else h1-=t[i]*5;
            int h2=h1,ans2=0,ans3[35];
            mem(ans3);
            P p1;
            for(int j=0; j<=i; j++)
            {
                p1.val=f[j];
                p1.poi=j;
                pq.push(p1);
            }

            while(h2-5>=0)
            {
                h2-=5;
                P p2=pq.top();
                if(p2.val<=0) break;
                else
                {
                    pq.pop();
                    ans2+=p2.val;
                    p2.val-=d[p2.poi];
                    ans3[p2.poi]++;
                    pq.push(p2);

                }

            }

            if(ans2>ans)
            {
                for(int i=0;i<n;i++)
                ans1[i]=ans3[i];
                ans=ans2;
            }
        while(!pq.empty()) pq.pop();

        }

        int summ=0;
        int endd;

        for(int i=0;i<n;i++)
        {
            summ+=ans1[i]*5;
            if(ans1[i]) endd=i;
        }
        summ=h-summ;
        for(int i=0;i<=endd;i++)
        {
            summ-=t[i]*5;
        }
        if(summ<0) summ=0;
        for(int i=0;i<n-1;i++)
        {
            if(i==0) cout<<ans1[i]*5+summ<<", ";
            else cout<<ans1[i]*5<<", ";
        }

        cout<<ans1[n-1]*5<<endl;

        cout<<"Number of fish expected: "<<ans<<endl;

        cout<<endl;
    }

    return 0;
}