AcWing 1262. 鱼塘钓鱼 多路归并

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

鱼塘编号	1	2	3	4	5
第1分钟能钓到的鱼的数量（1..1000）	10	14	20	16	9
每钓鱼1分钟钓鱼数的减少量（1..100)	2	4	6	5	3
当前鱼塘到下一个相邻鱼塘需要的时间（单位：分钟）	3	5	4	4

即：在第 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 <cstring>
#include <algorithm>

using namespace std;

const int N = 110;

int a[N], d[N], l[N], spend[N];

int get(int k)//判断这次的分钟一共钓了多少鱼
{
    return max(0, a[k] - d[k] * spend[k]);
}

int work(int n, int T)
{
    int res = 0;
    memset(spend, 0, sizeof spend);

    for (int i = 0; i < T; i ++ )
    {
        int t = 1;//第一个鱼塘
        for (int j = 2; j <= n; j ++ )//从第二个开始枚举，到最终能到哪一个位置
            if (get(t) < get(j))//判断第t分钟在哪个鱼塘钓的鱼最多就去那里掉
                t = j;

        res += get(t);//答案加上
        spend[t] ++ ;//记录在哪个t（鱼塘）钓了多少时间
    }

    return res;
}

int main()
{
    int n, T;
    scanf("%d", &n);

    for (int i = 1; i <= n; i ++ ) scanf("%d", &a[i]);
    for (int i = 1; i <= n; i ++ ) scanf("%d", &d[i]);
    for (int i = 2; i <= n; i ++ )
    {
        scanf("%d", &l[i]);
        l[i] += l[i - 1];//用前缀和来处理，看看最终走到了哪个鱼塘
    }
    scanf("%d", &T);

    int res = 0;
    for (int i = 1; i <= n; i ++ )
        res = max(res, work(i, T - l[i]));//总时间减去路上的时间就是钓鱼的时间

    printf("%d\n", res);
    return 0;
}