HashMap 提供了多种方法来操作键值对,除了常用的 put()、get() 和 remove() 方法之外,还有一些不太常用但非常有用的方法。下面是几个不常用但很有用的 HashMap 方法及其使用示例:
- compute(K key, BiFunction<? super K, ? super V, ? extends V> remappingFunction)
此方法允许你基于现有值(如果存在)或计算新值,并将其放入映射中。
示例:
java
import java.util.HashMap;
import java.util.function.BiFunction;
public class Main {
public static void main(String[] args) {
HashMap<String, Integer> map = new HashMap<>();
map.put("apple", 1);
// 如果有值,则增加1;如果没有,则设置为10
BiFunction<String, Integer, Integer> function = (key, oldValue) -> oldValue == null ? 10 : oldValue + 1;
map.compute("apple", function);
System.out.println(map.get("apple")); // 输出: 2
map.compute("banana", function);
System.out.println(map.get("banana")); // 输出: 10
}
}
- computeIfAbsent(K key, Function<? super K, ? extends V> mappingFunction)
当指定键没有对应值时,使用提供的映射函数计算其值并放入映射中。
示例:
java
import java.util.HashMap;
import java.util.function.Function;
public class Main {
public static void main(String[] args) {
HashMap<String, Integer> map = new HashMap<>();
Function<String, Integer> mappingFunction = key -> key.length();
Integer value = map.computeIfAbsent("apple", mappingFunction);
System.out.println(value); // 输出: 5
// 已存在的键不会重新计算
value = map.computeIfAbsent("apple", mappingFunction);
System.out.println(value); // 输出: 5
}
}
- computeIfPresent(K key, BiFunction<? super K, ? super V, ? extends V> remappingFunction)
只有在键对应的值非空时,才应用给定的重映射函数计算新值。
示例:
java
import java.util.HashMap;
import java.util.function.BiFunction;
public class Main {
public static void main(String[] args) {
HashMap<String, Integer> map = new HashMap<>();
map.put("apple", 1);
BiFunction<String, Integer, Integer> remappingFunction = (key, oldValue) -> oldValue + 1;
map.computeIfPresent("apple", remappingFunction);
System.out.println(map.get("apple")); // 输出: 2
// 对于不存在的键,不会执行任何操作
map.computeIfPresent("banana", remappingFunction);
System.out.println(map.containsKey("banana")); // 输出: false
}
}
- merge(K key, V value, BiFunction<? super V, ? super V, ? extends V> remappingFunction)
如果指定的键尚未与某个值关联或关联的值为 null,则将该键与给定的值关联;否则,使用给定的重映射函数重新计算该键的新值。
示例:
java
import java.util.HashMap;
import java.util.function.BiFunction;
public class Main {
public static void main(String[] args) {
HashMap<String, Integer> map = new HashMap<>();
map.put("apple", 1);
BiFunction<Integer, Integer, Integer> remappingFunction = Integer::sum;
map.merge("apple", 2, remappingFunction);
System.out.println(map.get("apple")); // 输出: 3
map.merge("banana", 5, remappingFunction);
System.out.println(map.get("banana")); // 输出: 5
}
}