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