每日一练_PAT_B1001

鲁宾逊先生有一只宠物猴，名叫多多。这天，他们两个正沿着乡间小路散步，突然发现路边的告示牌上贴着一张小小的纸条：“欢迎免费品尝我种的花生！——熊字”。
鲁宾逊先生和多多都很开心，因为花生正是他们的最爱。在告示牌背后，路边真的有一块花生田，花生植株整齐地排列成矩形网格。有经验的多多一眼就能看出，每棵花生植株下的花生有多少。为了训练多多的算术，鲁宾逊先生说:“你先找出花生最多的植株，去采摘它的花生;然后再找出剩下的植株里花生最多的，去采摘它的花生;依此类推，不过你一定要在我限定的时间内回到路边。”
我们假定多多在每个单位时间内,可以做下列四件事情中的一件：
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（没有花生），其他所有植株下花生的数目都不相同。

  输出描述:

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

import java.util.*;

public class Main {
    public static void main(String[] args) {
        int m, n, k;
        Scanner s = new Scanner(System.in);
        m = s.nextInt();//行
        n = s.nextInt();//列
        k = s.nextInt();//单位时间
        //先找到有花生的植株
        HashMap<Integer, Peanut> peanut = new HashMap<>();
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                int amount = s.nextInt();
                if (amount > 0)
                    peanut.put(amount, new Peanut(m, n, amount));
            }
        }
        //再通过花生数将植株排序
        Comparator<Integer> cmp = new Comparator<Integer>() {
            @Override
            public int compare(Integer integer1, Integer integer2) {
                return integer1 - integer2;
            }
        };
        ArrayList<Integer> list = new ArrayList<>(peanut.keySet());//取出花生数量集合
        if (list.isEmpty())
            System.out.println(0);
        else {
            Collections.sort(list, cmp);
            int index = 0;
            int key = list.get(index++);
            // 判断该花生能不能被采摘
            // 获取数量最多的那个花生的植株
            Peanut node = peanut.get(key);
            int val = 0; // 创建一个变量，存放能采摘到的花生数量
            // 计算移动时间，采摘时间，离开时间
            int moveTime = node.x + 1, takeTime = 1, leavlTime = node.x + 1;
            // 判断能不能采摘
            while (k - moveTime - takeTime - leavlTime >= 0) {
                // 可以采摘，减去移动时间，采摘时间
                k = k - moveTime - takeTime;
                // 累加采摘的花生数量
                val += node.amount;
                // 判断有无下一植株
                if (index >= list.size()) break;
                // 获取下一植株
                Peanut nextNode = peanut.get(list.get(index++));
                // 获取移动到下一植株的时间
                moveTime = Math.abs(node.x - nextNode.x)
                        + Math.abs(node.y - nextNode.y);
                // 获取从下一植株离开的时间
                leavlTime = nextNode.x + 1;
                node = nextNode;
            }
        }
    }
}
class Peanut{
    int x,y,amount;
    Peanut(int x , int y , int amount){
        this.x = x;
        this.y = y;
        this.amount = amount;
    }
}

想了大半天，最后还是借鉴了大佬的代码。这里的测试用例貌似有问题，怎么搞都通不过.....