[模拟过程]花生问题

1052 花生问题

题目描述

鲁宾逊先生有一只宠物猴，名叫多多。这天，他们两个正沿着乡间小路散步，突然发现路边的告示牌上贴着一张小小的纸条：“欢迎免费品尝我种的花生！！---熊先生”。
鲁宾逊先生和多多都很开心，因为花生正是他们的最爱。在告示牌背后，路边真的有一块花生田，花生植株整齐地排列成矩形网格（如图1）。有经验的多多一眼就能看出，每棵花生植株下的花生有多少。例如在图2中的花生田里，只有位于(2, 5), (3, 7), (4, 2), (5, 4)的植株下长有花生，个数分别为13, 7, 15, 9。我们假定多多在每个单位时间内，可以做下列四件事情中的一件：(1)从路边跳到最靠近路边（即第一行）的某棵花生植株；(2)从一棵植株跳到前后左右与之相邻的另一棵植株；(3)采摘一棵植株下的花生；(4)从最靠近路边（即第一行）的某棵花生植株跳回路边。
 

为了训练多多的算术，鲁宾逊先生说：“你先找出花生最多的植株，去采摘它的花生；然后再找出剩下的植株里花生最多的，去采摘它的花生；依此类推，不过你一定要在我限定的时间内回到路边。” 例如在图2中，沿着图示的路线，多多在21个单位时间内，最多可以采到37个花生。
现在给定一块花生田的大小和花生的分布，请问在限定时间内，多多最多可以采到多少个花生？注意可能只有部分植株下面长有花生，假设这些植株下的花生个数各不相同。

关于输入

输入第一行代表组数T。
之后每组第一行包括三个整数，M, N和K，用空格隔开；表示花生田的大小为M * N（1 <= M, N <= 20），多多采花生的限定时间为K（0 <= K <= 1000）个单位时间。接下来的M行，每行包括N个非负整数，也用空格隔开；第i + 1行的第j个整数Pij（0 <= Pij <= 500）表示花生田里植株(i, j)下花生的数目，0表示该植株下没有花生。

关于输出

输出包括T组，每一行代表一组，只包含一个整数，即在限定时间内，多多最多可以采到花生的个数。

例子输入

2
6 7 21
0 0 0 0 0 0 0
0 0 0 0 13 0 0
0 0 0 0 0 0 7
0 15 0 0 0 0 0
0 0 0 9 0 0 0
0 0 0 0 0 0 0
6 7 20
0 0 0 0 0 0 0
0 0 0 0 13 0 0
0 0 0 0 0 0 7
0 15 0 0 0 0 0
0 0 0 9 0 0 0
0 0 0 0 0 0 0

例子输出

37
28

解题分析

注意到，本题输入的数据中有些是无效的，我们只需要挑选有效的数据即可。我们先从最多的花生开始采摘起，然后依次进行从高到低的采摘花生的操作，我们可以每一步采摘的时候看看采摘完然后剩下的时间够不够回到路边，如果够，那就采摘，否则直接回到路边。

代码实现

#include <bits/stdc++.h>
using namespace std;
typedef struct peanut{
    int x,y;
    int count;
} peanut;
bool comp(peanut x,peanut y){
    if(x.count>y.count) return 1;
    else return 0;
}
int main(){
    int T; cin>>T;
    while(T--){
    int M,N,K; cin>>M>>N>>K;
    vector<peanut> p;
    for(int i=1;i<M+1;i++){
        for(int j=1;j<N+1;j++){
            int c; cin>>c;
            if(c){
               peanut temp;
               temp.x=i; temp.y=j;
               temp.count=c;
               p.push_back(temp); 
            }
        }
    }
    sort(p.begin(),p.end(),comp);
    int cur_x=p[0].x,cur_y=p[0].y;
    if((2*cur_x+1)>K){
        cout<<0<<endl; break;
    }
    int sum=0; sum+=p[0].count;
    K-=cur_x+1;
    for(int i=1;i<p.size();i++){
        if(K-(abs(p[i].x-cur_x)+abs(p[i].y-cur_y)+1+p[i].x)<0){
            cout<<sum<<endl; break;
        }
        else{
            K-=(abs(p[i].x-cur_x)+abs(p[i].y-cur_y)+1);
            sum+=p[i].count;
            cur_x=p[i].x; cur_y=p[i].y;
        }
    }
    }
    return 0;
}