AcWing 1262. 鱼塘钓鱼(贪心+堆+多路归并)

题目链接

有 N 个鱼塘排成一排，每个鱼塘中有一定数量的鱼，例如：N=5 时，如下表：

即：在第 1 个鱼塘中钓鱼第 1 分钟内可钓到 10 条鱼，第 2 分钟内只能钓到 8 条鱼，……，第 5 分钟以后再也钓不到鱼了。

从第 1 个鱼塘到第 2 个鱼塘需要 3 分钟，从第 2 个鱼塘到第 3 个鱼塘需要 5 分钟，……

给出一个截止时间 T，设计一个钓鱼方案，从第 1 个鱼塘出发，希望能钓到最多的鱼。

假设能钓到鱼的数量仅和已钓鱼的次数有关，且每次钓鱼的时间都是整数分钟。

输入格式
共 5 行，分别表示：

• 第 1 行为 N；

• 第 2 行为第 1 分钟各个鱼塘能钓到的鱼的数量，每个数据之间用一空格隔开；

• 第 3 行为每过 1 分钟各个鱼塘钓鱼数的减少量，每个数据之间用一空格隔开；

• 第 4 行为当前鱼塘到下一个相邻鱼塘需要的时间；

• 第 5 行为截止时间 T。

输出格式

一个整数（不超过231−1），表示你的方案能钓到的最多的鱼。

数据范围

1≤N≤100,
1≤T≤1000

输入样例：

5
10 14 20 16 9
2 4 6 5 3
3 5 4 4
14

输出样例：

76

答案：

#include <iostream>
#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f3f3f3f
#define rep(i,a,b) for(auto i=a;i<=b;++i)
#define bep(i,a,b) for(auto i=a;i>=b;--i)
#define lowbit(x) x&(-x)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define PI acos(-1)
#define pb push_back
#define eps 1e-6
const int mod = 1e9 + 7;
const int N = 1e6 + 10;
const int M = 211;
int dx[]={-1, 0, 1, 0};
int dy[]={0, 1, 0, -1};
using namespace std;

int dp[M];
int mp[M];
int st[M];

void solve(){
    int n;
    cin>>n;
    rep(i,1,n) cin>>dp[i];
    rep(i,1,n) cin>>mp[i];
    ///求间隔时间前缀和
    rep(i,2,n){
        cin>>st[i];
        st[i]+=st[i-1];
    }
    int k;
    cin>>k;
    int maxn=-1;
    ///枚举只在前i个池塘钓鱼
    for(int i=1;i<=n&&st[i]<=k;i++){
        priority_queue<PII> que;
        rep(j,1,i) que.push({dp[j],mp[j]});
        int t=k-st[i];
        int res=0;
        ///枚举除间隔外在钓鱼的每一分钟
        rep(j,1,t){
            auto p=que.top();
            que.pop();
            if(p.first<=0) break;
            res+=p.first;
            p.first-=p.second; ///每分钟减少
            que.push(p); ///减少后入堆
        }
        maxn=max(maxn,res);
    }
    cout<<maxn<<endl;
}

int main() {
    solve();
	return 0;
}