noip2009-4 道路游戏

题目描述

小新正在玩一个简单的电脑游戏。
游戏中有一条环形马路，马路上有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个单位时间内，扣除购买机器人
花费的金币之后，小新最多能收集到多少金币。

样例输入

2 3 2 1 2 3 2 3 4 1 2

样例输出

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。

初中时候的神题，现在看来，也不过如此。

我先写了一个会超时的，待会儿我会讲如何改进

#include<cstdio>
#include<iostream>
using namespace std;
int f[1005],b[1005],a[1005][1005];
int n,m,p,x;
int work(int l,int r,int id)
{
    int s=0,y=id;
    for(int i=l;i<=r;i++) 
    {
        s=s+a[y][i];
        if(y==n) y=1;else y++;
    }
    return s;
}   
int main()
{
    cin>>n>>m>>p;
    for(int i=1;i<=n;i++) 
    for(int j=1;j<=m;j++) scanf("%d",&a[i][j]);
    for(int i=1;i<=n;i++) scanf("%d",&b[i]);
    for(int i=1;i<=m;i++) f[i]=-1000000000;  
    for(int i=1;i<=m;i++) //枚举时间
    for(int j=1;j<=n;j++) //枚举上一次的出发点
    {
        x=min(i,p);
        for(int k=1;k<=x;k++) //枚举上一次的步数
        f[i]=max(f[i-k]-b[j]+work(i-k+1,i,j),f[i]); //每次要计算上一次积攒的金币
    }
    cout<<f[m];   
    return 0;
}

显而易见，目前唯一的突破口是计算价值，想一想，我们有没有必要每一次都算一遍呢？

本题因为有“时间”这个量，导致我们很难用前缀和来直接计算，如果我们想顺带累加价值，我们该怎么办呢？

原来枚举状态的时候，上一次的起点和终点都是不确定的，我们如果可以锁定一个，那么就可以在枚举步数的时候累加了，so，请看下面的代码

#include<cstdio>
#include<iostream>
using namespace std;
int f[1005],b[1005],a[1005][1005];
int n,m,p,x,y,s;
int main()
{
    cin>>n>>m>>p;
    for(int i=1;i<=n;i++) 
    for(int j=1;j<=m;j++) scanf("%d",&a[i][j]);
    for(int i=1;i<=n;i++) scanf("%d",&b[i]);
    for(int i=1;i<=m;i++) f[i]=-1000000000;  
    for(int i=1;i<=m;i++) 
    for(int j=1;j<=n;j++) //枚举i时的终点
    {
        x=min(i,p);
        y=j-1;
        if(y==0) y=n;
        s=a[y][i];
        for(int k=1;k<=x;k++) 
        {
            f[i]=max(f[i-k]-b[y]+s,f[i]);
            if(y==1) y=n;else y--;
            s=s+a[y][i-k];  //每次顺带累加
        }
    }
    cout<<f[m];   
    return 0;
}