洛谷 P1070 道路游戏(动态规划)

题目描述

小新正在玩一个简单的电脑游戏。

游戏中有一条环形马路，马路上有 n 个机器人工厂，两个相邻机器人工厂之间由一小段马路连接。小新以某个机器人工厂为起点，按顺时针顺序依次将这 n 个机器人工厂编号为1~n，因为马路是环形的，所以第 n 个机器人工厂和第 1 个机器人工厂是由一段马路连接在一起的。小新将连接机器人工厂的这 n 段马路也编号为 1~n，并规定第 i 段马路连接第 i 个机器人工厂和第 i+1 个机器人工厂（1≤i≤n-1），第 n 段马路连接第 n 个机器人工厂和第 1个机器人工厂。

游戏过程中，每个单位时间内，每段马路上都会出现一些金币，金币的数量会随着时间发生变化，即不同单位时间内同一段马路上出现的金币数量可能是不同的。小新需要机器人的帮助才能收集到马路上的金币。所需的机器人必须在机器人工厂用一些金币来购买，机器人一旦被购买，便会沿着环形马路按顺时针方向一直行走，在每个单位时间内行走一次，即从当前所在的机器人工厂到达相邻的下一个机器人工厂，并将经过的马路上的所有金币收集给小新，例如，小新在 i（1≤i≤n）号机器人工厂购买了一个机器人，这个机器人会从 i 号机器人工厂开始，顺时针在马路上行走，第一次行走会经过 i 号马路，到达 i+1 号机器人工厂（如果 i=n，机器人会到达第 1 个机器人工厂），并将 i 号马路上的所有金币收集给小新。 游戏中，环形马路上不能同时存在 2 个或者 2 个以上的机器人，并且每个机器人最多能够在环形马路上行走 p 次。小新购买机器人的同时，需要给这个机器人设定行走次数，行走次数可以为 1~p 之间的任意整数。当马路上的机器人行走完规定的次数之后会自动消失，小新必须立刻在任意一个机器人工厂中购买一个新的机器人，并给新的机器人设定新的行走次数。

以下是游戏的一些补充说明：

1.游戏从小新第一次购买机器人开始计时。

2.购买机器人和设定机器人的行走次数是瞬间完成的，不需要花费时间。

3.购买机器人和机器人行走是两个独立的过程，机器人行走时不能购买机器人，购买完机器人并且设定机器人行走次数之后机器人才能行走。

4.在同一个机器人工厂购买机器人的花费是相同的，但是在不同机器人工厂购买机器人的花费不一定相同。

5.购买机器人花费的金币，在游戏结束时再从小新收集的金币中扣除，所以在游戏过程中小新不用担心因金币不足，无法购买机器人而导致游戏无法进行。也因为如此，游戏结束后，收集的金币数量可能为负。

现在已知每段马路上每个单位时间内出现的金币数量和在每个机器人工厂购买机器人需要的花费，请你告诉小新，经过 m 个单位时间后，扣除购买机器人的花费，小新最多能收集到多少金币。

---

输入输出格式

输入格式：

第一行 3 个正整数，n，m，p，意义如题目所述。

接下来的 n 行，每行有 m 个正整数，每两个整数之间用一个空格隔开，其中第 i 行描

述了 i 号马路上每个单位时间内出现的金币数量（1≤金币数量≤100），即第 i 行的第 j（1≤j≤m）个数表示第 j 个单位时间内 i 号马路上出现的金币数量。

最后一行，有 n 个整数，每两个整数之间用一个空格隔开，其中第 i 个数表示在 i 号机器人工厂购买机器人需要花费的金币数量（1≤金币数量≤100）。

输出格式：

共一行，包含 1 个整数，表示在 m 个单位时间内，扣除购买机器人

花费的金币之后，小新最多能收集到多少金币。

---

输入输出样例

输入样例#1：

2 3 2
1 2 3
2 3 4
1 2

输出样例#1：

5

---

说明

【数据范围】

对于 40%的数据，2≤n≤40，1≤m≤40。

对于 90%的数据，2≤n≤200，1≤m≤200。

对于 100%的数据，2≤n≤1000，1≤m≤1000，1≤p≤m。

NOIP 2009 普及组 第四题

---

Solution

这题一开始我很快想出了$O(n^3)$的做法，但苦苦思索不出来如何用单调队列降成$O(n^2)$，我们发现每一轮后购买机器人的位置是任意的，这成了本题最有趣的特征，也是难点。我们需要开两个数组，分别记两个状态，$f[i][j]$代表第$i$秒在第$j$个工厂，然后$F[i]$代表第$i$秒过去后的全局最优值(我们发现给定时间的最优值是一定的，即与位置无关)。然后就可以很快推出需要预处理的$O(n^3)$的表达式。然而预处理我想不到很快求出一段道路以及一段时间的金币的部分和，后来我终于知道在$n^2$内求出来，然后再单调队列优化上又较难实现。我们需要开$N$个单调队列维护$F[i]$，$sum$，$cost$等，比较难写。

而后其实我把题目想复杂了，这题其实可以不用单调队列优化，非常好懂的，我们利用$dp$加上些许贪心思想，同样记$f[i][j]$,$F[i]$，然后将问题拆分成两种情况:

一种是当前第$i$秒在第$j$个点，我们上一个点不新买机器人，那么就由上一个$f[i-1][pre[j]]$加上第$pre[j]$条路上的金币转移就行了，步数的限制就多开一个数组限制步数并更新。

另一种是上一次新买机器人，这样由于任意性，$F[i-1]$秒的价值得到后就可立刻去上一个地方买，减去$cost$加上路上金币，最后就将$step$清为1，$F[i]$在求$f[i][j]$时一起求，上一秒$F[i-1]$已求出了。于是$O(n^2)$，不用任何优化。

值得注意的是，我们以时间为阶段，决定上一个点买不买机器人，比决定当前点买不买要更好处理和转移。

这种方法的核心是这一秒只跟上一秒有关，不考虑再之前的情况，使过程大大简化，省去了部分和，单调队列的操作，简洁易懂。以后思考问题不要将其想得太复杂，尽量简化，寻找最简洁的路径才是。

---

Code

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#define N 1111
#define oo 0x7fffffff

using namespace std;

int n, m, p, coin[N][N];
int cost[N], f[N][N], F[N], pre[N], step[N][N];

int main(){

    scanf("%d%d%d", &n, &m, &p);

    for(int i = 1; i <= n; i++)
     for(int j = 1; j <= m; j++)
       scanf("%d", &coin[i][j]);

    for(int i = 1; i <= n; i++)
      scanf("%d", &cost[i]);

    for(int i = 1; i <= m; i++)  F[i] = -oo;
    for(int i = 1; i <= m; i++)
     for(int j = 1; j <= n; j++)
       f[i][j] = -oo;

     for(int i = 1; i <= n; i++)  pre[i] = i != 1 ? i - 1 : n, f[0][i] = 0; 
     F[0] = 0;

    for(int i = 1; i <= m; i++)
     for(int j = 1; j <= n; j++){
       if(step[i-1][pre[j]] < p && i > 1 && f[i-1][pre[j]] + coin[pre[j]][i] > f[i][j]){
         f[i][j] = f[i-1][pre[j]] + coin[pre[j]][i];
         step[i][j] = step[i-1][pre[j]] + 1;
       }
       if(f[i][j] < F[i-1] + coin[pre[j]][i] - cost[pre[j]]){
         f[i][j] = F[i-1] + coin[pre[j]][i] - cost[pre[j]];
         step[i][j] = 1;
       }
       F[i] = max(F[i], f[i][j]);
     }

    printf("%d", F[m]);
    return 0;
}

---

在我看不见星星的时候，星星也看不见我。