题解 | #奶牛排队问题# java
奶牛排队问题
https://www.nowcoder.com/practice/e974a33a14a344278c3a359a1ebf833e
import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param weights int整型一维数组 * @param n int整型 * @param k int整型 * @return bool布尔型 */ public boolean checkDuplicate (int[] weights, int n, int k) { // write code here Map<Integer, Integer> map = new HashMap<>(); for (int i = 0; i < n; i++) { // 如果当前体重已经在map中存在,并且当前位置与最近一次出现的位置的距离不超过k,则返回true if (map.containsKey(weights[i]) && (i - map.get(weights[i])) <= k) { return true; } // 更新当前体重在map中最新的位置 map.put(weights[i], i); } return false; } }
Java语言。
该题考察的知识点是使用HashMap来进行元素的查找和插入,以及遍历数组。
代码的文字解释如下:
- 创建一个HashMap对象map,用于存储奶牛的体重和最新出现的位置。
- 使用for循环遍历奶牛队列,从第一个奶牛到最后一个奶牛。
- 在循环中,首先判断当前奶牛的体重是否已经在map中存在。如果存在,说明之前已经遇到过具有相同体重的奶牛。
- 接着判断当前奶牛的位置与最近一次出现的位置的距离是否不超过k。如果满足这个条件,说明找到了符合要求的两头奶牛,直接返回true。
- 如果不满足上述两个条件,则将当前奶牛的体重和位置存入map中,以便后续的查找。
- 循环结束后,如果没有找到符合要求的两头奶牛,返回false。