[NOIP2009 普及组] 道路游戏（含代码）

[NOIP2009 普及组] 道路游戏

题目描述

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

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

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

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

1. 游戏从小新第一次购买机器人开始计时。
2. 购买机器人和设定机器人的行走次数是瞬间完成的，不需要花费时间。
3. 购买机器人和机器人行走是两个独立的过程，机器人行走时不能购买机器人，购买完机器人并且设定机器人行走次数之后机器人才能行走。
4. 在同一个机器人工厂购买机器人的花费是相同的，但是在不同机器人工厂购买机器人的花费不一定相同。
5. 购买机器人花费的金币，在游戏结束时再从小新收集的金币中扣除，所以在游戏过程中小新不用担心因金币不足，无法购买机器人而导致游戏无法进行。也因为如此，游戏结束后，收集的金币数量可能为负。

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

输入格式

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

接下来的 $n$ 行，每行有 $m$ 个正整数，每两个整数之间用一个空格隔开，其中第 $i$ 行描述了 $i$ 号马路上每个单位时间内出现的金币数量（$1\le $ 金币数量 $\le 100$），即第 $i$ 行的第 $j$（$1\le j\le m$）个数表示第 $j$ 个单位时间内 $i$ 号马路上出现的金币数量。

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

输出格式

共一行，包含 $1$ 个整数，表示在 $m$ 个单位时间内，扣除购买机器人花费的金币之后，小新最多能收集到多少金币。

样例 #1

样例输入 #1

2 3 2 
1 2 3 
2 3 4 
1 2

样例输出 #1

5

提示

对于 $40\%$ 的数据，$2\le n\le 40$，$1\le m\le 40$。

对于 $90\%$ 的数据，$2\le n\le 200$，$1\le m\le 200$。

对于 $100\%$ 的数据，$2\le n\le 1000$，$1\le m\le 1000$，$1\le p\le m$。

题目来源

NOIP 2009 普及组 第四题（洛谷）

题解

#include <iostream>
#include <cstring>
using namespace std;

int a[1005][1005], cost[1005];
int dp[1005];

int main() {
    int n, m, p;
    cin >> n >> m >> p;
    
    for (int i = 1; i <= n; i++) 
        for (int j = 1; j <= m; j++) 
            cin >> a[i][j];
    
    for (int i = 1; i <= n; i++) 
        cin >> cost[i];
    
    memset(dp, ~0x7f, sizeof(dp));
    dp[0] = 0;  // 初始化
    
    for (int i = 1; i <= m; i++) {  // 枚举时间，前i分钟
        for (int j = 1; j <= n; j++) {  // 枚举结束位置，以j号工厂结束
            int sum = 0;  // sum表示走过来的过程中能收集多少金币
            for (int k = 1; k <= p && k <= i; k++) {
                // k表示走了k步到达j号工厂，同时，也表示走到的用时，所以有 k<=p && k<=i
                int last;  // last表示走k步前，工厂是多少号
                if (j - k <= 0) 
                    last = n + (j - k) % n;
                else 
                    last = j - k;   // 环形道路
                
                sum += a[last][i - k + 1];  // 求和
                dp[i] = max(dp[i], dp[i - k] + sum - cost[last]);
                // 状态转移方程，第i分钟的最大值=max(前i-k分钟最大值+走过来得到金币数)
            }
        }
    }
    
    cout << dp[m];  // 第m分钟的最大值，即答案
    return 0;
}