[PAT乙级]自测3

自测题4 1001.采花生

题目描述
鲁宾逊先生有一只宠物猴，名叫多多。这天，他们两个正沿着乡间小路散步，突然发现路边的告示牌上贴着一张小小的纸条：“欢迎免费品尝我种的花生！——熊字”。

鲁宾逊先生和多多都很开心，因为花生正是他们的最爱。在告示牌背后，路边真的有一块花生田，花生植株整齐地排列成矩形网格。有经验的多多一眼就能看出，每棵花生植株下的花生有多少。为了训练多多的算术，鲁宾逊先生说:“你先找出花生最多的植株，去采摘它的花生;然后再找出剩下的植株里花生最多的，去采摘它的花生;依此类推，不过你一定要在我限定的时间内回到路边。”

我们假定多多在每个单位时间内,可以做下列四件事情中的一件：

1. 从路边跳到最靠近路边（即第一行）的某棵花生植株；
2. 从一棵植株跳到前后左右与之相邻的另一棵植株；
3. 采摘一棵植株下的花生；
4. 从最靠近路边（即第一行）的某棵花生植株跳回路边。

现在给定一块花生田的大小和花生的分布，请问在限定时间内，多多最多可以采到多少个花生？

注意可能只有部分植株下面长有花生，假设这些植株下的花生个数各不相同。例如花生田里只有位于(2, 5), (3, 7), (4, 2), (5, 4)的植株下长有花生，个数分别为 13, 7, 15, 9。多多在 21 个单位时间内，只能经过(4, 2)、(2, 5)、(5, 4)，最多可以采到 37 个花生。

输入描述:
输入包含多组数据，每组数据第一行包括三个整数 M（1≤M≤20）、N（1≤N≤20）和 K（0≤K≤1000），用空格隔开；表示花生田的大小为 M * N，多多采花生的限定时间为 K个单位时间。

紧接着 M 行，每行包括 N 个自然数 P（0≤P≤500），用空格隔开；表示花生田里植株下花生的数目，并且除了0（没有花生），其他所有植株下花生的数目都不相同。

输出描述:
对应每一组数据，输出一个整数，即在限定时间内，多多最多可以采到花生的个数。

输入例子:

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

输出例子:

37

• 这道题严格按照题意走，从最多花生的那一棵开始采摘
• 不需要最优解！！！！！！！
• 提交代码时有个样例一直不通过，样例只有一个1，感觉很无奈啊，明明输入至少三个数，这让我怎么破，很绝望
• 提示是这样的

测试用例:
1
对应输出应该为:
0
你的输出为:
Q

没有AC的代码：(我实在找不出错误)

#include <iostream>
#include <vector>
#include <cstdio>
using namespace std;
struct node{
    int x;
    int y;
};
void sort(vector<int> &v, vector<node> &pos){
    int vlen = v.size();
    for(int i=0; i<vlen-1; i++){
        for(int j=i+1; j<vlen; j++){
            if(v[i] < v[j]){
                int temp = v[i];
                v[i] = v[j];
                v[j] = temp;

                node temp1 = pos[i];
                pos[i] = pos[j];
                pos[j] = temp1;
            }
        }
    }
}
int len(node a, node b){
    int m=0, n=0;
    m = a.x-b.x>0 ? a.x-b.x : b.x-a.x;
    n = a.y-b.y>0 ? a.y-b.y : b.y-a.y;
    return m+n;
}
int main()
{
    int m,n,k;
    while(~scanf("%d %d %d", &m, &n, &k)){
        if(k == 0){
            cout << 0;
            return 0;
        }
        int arr[m+1][n+1];
        vector<int> v;//地下有花生的个数
        vector<node> pos;//地下有花生的坐标
        for(int i=1; i<=m; i++){
            for(int j=1; j<=n; j++){
                cin >> arr[i][j];
                if(arr[i][j] != 0){
                    v.push_back(arr[i][j]);
                    node temp;
                    temp.x = i;
                    temp.y = j;
                    pos.push_back(temp);
                }
            }
        }
        if(v.size() == 0){
            cout << 0;
            return 0;
        }
        sort(v, pos);//按照采摘顺序（花生数目从大到小）排序
        int res=0;//最大采摘数目
        node a; a.x = 0; a.y = pos[0].y;//起始位置
        int vlen = v.size();
        for(int i=0; i<vlen; i++){
            if(k >= len(pos[i],a)+1+pos[i].x){
                res += v[i];
                k -= len(pos[i],a)+1;//采摘时间减少
                a.x = pos[i].x;//获取新的起始位置
                a.y = pos[i].y;
            }else{
                break;
            }
        }
        cout << res;
    }

    return 0;
}

注意：以上答案没有AC，我看了讨论区，其实基本都是一个思路，有兴趣点击链接