在本教程中,我們將借助示例學(xué)習(xí)Java TreeMap類及其操作。
Java 集合框架的TreeMap類提供了樹形數(shù)據(jù)結(jié)構(gòu)實現(xiàn)。
它繼承了NavigableMap接口。

為了創(chuàng)建TreeMap,我們必須首先導(dǎo)入java.util.TreeMap包。 導(dǎo)入程序包后,可以使用以下方法在Java中創(chuàng)建TreeMap。
TreeMap<Key, Value> numbers = new TreeMap<>();
在上面的代碼中,我們創(chuàng)建了一個沒有任何參數(shù)的名為numbers的TreeMap。在本示例中,TreeMap中的元素是自然排序的(升序)。
但是,我們可以通過使用Comparator接口自定義元素的排序。我們將在本教程的后面部分中學(xué)習(xí)它。
這里,
Key - 用于關(guān)聯(lián)map中每個元素(值)的唯一標(biāo)識符
Value - map中與鍵相關(guān)聯(lián)的元素
TreeMap類提供了各種方法,允許我們對映射執(zhí)行操作。
put() - 將指定的鍵/值映射(條目)插入到映射中
putAll() - 將指定映射中的所有條目插入到此映射中
putIfAbsent() - 如果映射中不存在指定的鍵,則將指定的鍵/值映射插入到map中
例如,
import java.util.TreeMap;
class Main {
public static void main(String[] args) {
//創(chuàng)建偶數(shù)的TreeMap
TreeMap<String, Integer> evenNumbers = new TreeMap<>();
// 使用 put()
evenNumbers.put("Two", 2);
evenNumbers.put("Four", 4);
// 使用 putIfAbsent()
evenNumbers.putIfAbsent("Six", 6);
System.out.println("偶數(shù)的TreeMap: " + evenNumbers);
//Creating TreeMap of numbers
TreeMap<String, Integer> numbers = new TreeMap<>();
numbers.put("One", 1);
// 使用 putAll()
numbers.putAll(evenNumbers);
System.out.println("TreeMap 的數(shù)字: " + numbers);
}
}輸出結(jié)果
偶數(shù)的TreeMap: {Four=4, Six=6, Two=2}
TreeMap 的數(shù)字: {Four=4, One=1, Six=6, Two=2}1.使用entrySet(),keySet()和values()
entrySet() - 返回TreeMap的所有鍵/值映射(條目)的集合
keySet() - 返回TreeMap的所有鍵的集合
values() - 返回TreeMap的所有圖的集合
例如,
import java.util.TreeMap;
class Main {
public static void main(String[] args) {
TreeMap<String, Integer> numbers = new TreeMap<>();
numbers.put("One", 1);
numbers.put("Two", 2);
numbers.put("Three", 3);
System.out.println("TreeMap: " + numbers);
// 使用 entrySet()
System.out.println("Key/Value 映射: " + numbers.entrySet());
// 使用 keySet()
System.out.println("Keys: " + numbers.keySet());
// 使用 values()
System.out.println("Values: " + numbers.values());
}
}輸出結(jié)果
TreeMap: {One=1, Three=3, Two=2}
Key/Value 映射: [One=1, Three=3, Two=2]
Keys: [One, Three, Two]
Values: [1, 3, 2]2.使用get() 和 getOrDefault()
get() - 返回與指定鍵關(guān)聯(lián)的值。如果找不到鍵,則返回null。
getOrDefault() - 返回與指定鍵關(guān)聯(lián)的值。如果找不到鍵,則返回指定的默認(rèn)值。
例如,
import java.util.TreeMap;
class Main {
public static void main(String[] args) {
TreeMap<String, Integer> numbers = new TreeMap<>();
numbers.put("One", 1);
numbers.put("Two", 2);
numbers.put("Three", 3);
System.out.println("TreeMap: " + numbers);
// 使用 get()
int value1 = numbers.get("Three");
System.out.println("使用 get(): " + value1);
// 使用 getOrDefault()
int value2 = numbers.getOrDefault("Five", 5);
System.out.println("使用 getOrDefault(): " + value2);
}
}輸出結(jié)果
TreeMap: {One=1, Three=3, Two=2}
使用 get(): 3
使用 getOrDefault(): 5在這里,getOrDefault()方法沒有找到鍵Five。因此,它返回指定的默認(rèn)值5。
remove(key) - 返回并從TreeMap中刪除與指定鍵關(guān)聯(lián)的條目
remove(key, value) -僅當(dāng)指定鍵與指定值相關(guān)聯(lián)時才從映射中刪除條目,并返回布爾值
例如,
import java.util.TreeMap;
class Main {
public static void main(String[] args) {
TreeMap<String, Integer> numbers = new TreeMap<>();
numbers.put("One", 1);
numbers.put("Two", 2);
numbers.put("Three", 3);
System.out.println("TreeMap: " + numbers);
//具有單參數(shù)刪除方法
int value = numbers.remove("Two");
System.out.println("被刪除的值: " + value);
//具有兩個參數(shù)的刪除方法
boolean result = numbers.remove("Three", 3);
System.out.println("條目 {Three=3} 被刪除? " + result);
System.out.println("更新后的TreeMap: " + numbers);
}
}輸出結(jié)果
TreeMap: {One=1, Three=3, Two=2}
被刪除的值 = 2
條目 {Three=3} 被刪除? True
更新后的TreeMap: {One=1}replace(key, value)-用key新的替換指定映射的值value
replace(key, old, new) -僅當(dāng)舊值已與指定鍵關(guān)聯(lián)時,才用新值替換舊值
replaceAll(function) -用指定的結(jié)果替換map的每個值 function
例如,
import java.util.TreeMap;
class Main {
public static void main(String[] args) {
TreeMap<String, Integer> numbers = new TreeMap<>();
numbers.put("First", 1);
numbers.put("Second", 2);
numbers.put("Third", 3);
System.out.println("Original TreeMap: " + numbers);
// 使用 replace()
numbers.replace("Second", 22);
numbers.replace("Third", 3, 33);
System.out.println("TreeMap使用replace()方法: " + numbers);
// 使用 replaceAll()
numbers.replaceAll((key, oldValue) -> oldValue + 2);
System.out.println("TreeMap使用replaceAll()方法: " + numbers);
}
}輸出結(jié)果
Original TreeMap: {First=1, Second=2, Third=3}
TreeMap使用replace()方法: {First=1, Second=22, Third=33}
TreeMap使用replaceAll()方法: {First=3, Second=24, Third=35}在上述程序中,注意語句
numbers.replaceAll((key, oldValue) -> oldValue + 2);
在這里,我們傳遞了一個lambda表達(dá)式作為參數(shù)。
replaceAll()方法訪問map的所有條目。然后,它將所有元素替換為新值(從lambda表達(dá)式返回)。
因為TreeMap類實現(xiàn)了NavigableMap,所以它提供了在TreeMap元素上導(dǎo)航的各種方法。
firstKey() - 返回map的第一個鍵
firstEntry() - 返回映射的第一個鍵的鍵/值映射
lastKey() - 返回map的最后一個鍵
lastEntry() - 返回映射的最后一個鍵的鍵/值映射
例如,
import java.util.TreeMap;
class Main {
public static void main(String[] args) {
TreeMap<String, Integer> numbers = new TreeMap<>();
numbers.put("First", 1);
numbers.put("Second", 2);
numbers.put("Third", 3);
System.out.println("TreeMap: " + numbers);
// 使用 the firstKey() 方法
String firstKey = numbers.firstKey();
System.out.println("第一個鍵: " + firstKey);
// 使用 the lastKey() 方法
String lastKey = numbers.lastKey();
System.out.println("最后一個鍵: " + lastKey);
// 使用 firstEntry() 方法
System.out.println("第一項: " + numbers.firstEntry());
// 使用 the lastEntry() 方法
System.out.println("最后一項: " + numbers.lastEntry());
}
}輸出結(jié)果
TreeMap: {First=1, Second=2, Third=3}
第一個鍵: First
最后一個鍵: Third
第一項: First=1
最后一項: Third=3HigherKey() - 返回大于指定鍵的那些鍵中的最小的鍵。
HigherEntry() - 返回與所有大于指定鍵的鍵中最小的鍵相關(guān)的條目。
lowerKey() - 返回所有小于指定鍵的最大鍵。
lowerEntry() - 返回與所有小于指定鍵的鍵中最大的鍵關(guān)聯(lián)的條目。
ceilingKey() - 返回大于指定鍵的那些鍵中的最小的鍵。如果映射中存在作為參數(shù)傳遞的鍵,則它將返回該鍵。
ceilingEntry() - 返回與大于指定鍵的那些鍵中最小的鍵相關(guān)的條目。如果映射中存在與傳遞給自變量的鍵關(guān)聯(lián)的條目,則返回與該鍵關(guān)聯(lián)的條目。
floorKey() - 返回小于指定鍵的那些鍵中最大的鍵。如果存在作為參數(shù)傳遞的鍵,它將返回該鍵。
floorEntry() - 返回與小于指定鍵的那些鍵中最大的鍵相關(guān)的條目。如果存在作為參數(shù)傳遞的鍵,它將返回該鍵。
例如,
import java.util.TreeMap;
class Main {
public static void main(String[] args) {
TreeMap<String, Integer> numbers = new TreeMap<>();
numbers.put("First", 1);
numbers.put("Second", 5);
numbers.put("Third", 4);
numbers.put("Fourth", 6);
System.out.println("TreeMap: " + numbers);
// 使用 higher()
System.out.println("使用 higherKey(): " + numbers.higherKey("Fourth"));
System.out.println("使用 higherEntry(): " + numbers.higherEntry("Fourth"));
// 使用 lower()
System.out.println("\n使用 lowerKey(): " + numbers.lowerKey("Fourth"));
System.out.println("使用 lowerEntry(): " + numbers.lowerEntry("Fourth"));
// 使用 ceiling()
System.out.println("\n使用 ceilingKey(): " + numbers.ceilingKey("Fourth"));
System.out.println("使用 ceilingEntry(): " + numbers.ceilingEntry("Fourth"));
// 使用 floor()
System.out.println("\n使用 floorKey(): " + numbers.floorKey("Fourth"));
System.out.println("使用 floorEntry(): " + numbers.floorEntry("Fourth"));
}
}輸出結(jié)果
TreeMap: {First=1, Fourth=6, Second=5, Third=4}
使用 higherKey(): Second
使用 higherEntry(): Second=5
使用 lowerKey(): First
使用 lowerEntry(): First=1
使用 ceilingKey(): Fourth
使用 ceilingEntry(): Fourth=6
使用 floorkey(): Fourth
使用 floorEntry(): Fourth=6pollFirstEntry() - 返回并刪除與映射的第一個鍵關(guān)聯(lián)的條目
pollLastEntry() -返回并刪除與映射的最后一個鍵關(guān)聯(lián)的條目
例如,
import java.util.TreeMap;
class Main {
public static void main(String[] args) {
TreeMap<String, Integer> numbers = new TreeMap<>();
numbers.put("First", 1);
numbers.put("Second", 2);
numbers.put("Third", 3);
System.out.println("TreeMap: " + numbers);
//使用 the pollFirstEntry() 方法
System.out.println("使用 pollFirstEntry(): " + numbers.pollFirstEntry());
// 使用 the pollLastEntry() 方法
System.out.println("使用 pollLastEntry(): " + numbers.pollLastEntry());
System.out.println("更新后的TreeMap: " + numbers);
}
}輸出結(jié)果
TreeMap: {First=1, Second=2, Third=3}
使用 pollFirstEntry(): First=1
使用 pollLastEntry(): Third=3
更新后的TreeMap: {Second=2}headMap(key,booleanValue)
headMap()方法返回指定鍵Key(作為參數(shù)傳遞)之前treemap的所有鍵/值對。
booleanValue參數(shù)是可選的。默認(rèn)值為false。
如果booleanValue為true,則該方法還包括指定key的鍵/值對。
例如,
import java.util.TreeMap;
class Main {
public static void main(String[] args) {
TreeMap<String, Integer> numbers = new TreeMap<>();
numbers.put("First", 1);
numbers.put("Second", 2);
numbers.put("Third", 3);
numbers.put("Fourth", 4);
System.out.println("TreeMap: " + numbers);
System.out.println("\n使用 headMap() 方法:");
// headMap() 使用默認(rèn) booleanValue為false
System.out.println("沒有指定布爾值: " + numbers.headMap("Fourth"));
// headMap()使用指定 booleanValue 為 true
System.out.println("指定布爾值為true: " + numbers.headMap("Fourth", true));
}
}輸出結(jié)果
TreeMap: {First=1, Fourth=4, Second=2, Third=3}
使用 headMap() 方法:
沒有指定布爾值: {First=1}
指定布爾值為true: {First=1, Fourth=4}tailMap(key,booleanValue)
tailMap()方法從指定鍵(作為參數(shù)傳遞)開始返回樹圖的所有鍵/值對。
booleanValue是一個可選的參數(shù)。默認(rèn)值為true。
如果booleanValue為false,則該方法不包含指定key的鍵/值對。
例如,
import java.util.TreeMap;
class Main {
public static void main(String[] args) {
TreeMap<String, Integer> numbers = new TreeMap<>();
numbers.put("First", 1);
numbers.put("Second", 2);
numbers.put("Third", 3);
numbers.put("Fourth", 4);
System.out.println("TreeMap: " + numbers);
System.out.println("\n使用 tailMap() 方法:");
// tailMap() booleanValue使用默認(rèn)值true
System.out.println("booleanValue使用默認(rèn)true: " + numbers.tailMap("Second"));
// tailMap() booleanValue 使用指定值false
System.out.println("booleanValue使用指定false : " + numbers.tailMap("Second", false));
}
}輸出結(jié)果
TreeMap: {First=1, Fourth=4, Second=2, Third=3}
使用 tailMap() 方法:
booleanValue使用默認(rèn)true: {Second=2, Third=3}
booleanValue使用指定false: {Third=3}subMap(k1,bV1,k2,bV2)
subMap()方法返回與k1和k2之間的鍵相關(guān)聯(lián)的所有條目,包括k1的條目。
bV1和bV2是可選的布爾參數(shù)。 bV1的默認(rèn)值為true,bV2的默認(rèn)值為false。
如果bV1為false,則該方法返回與k1和k2之間的鍵關(guān)聯(lián)的所有條目,但不包括k1的條目。
如果bV2為true,則該方法返回與k1和k2之間的鍵關(guān)聯(lián)的所有條目,包括k2的條目。
例如,
import java.util.TreeMap;
class Main {
public static void main(String[] args) {
TreeMap<String, Integer> numbers = new TreeMap<>();
numbers.put("First", 1);
numbers.put("Second", 2);
numbers.put("Third", 3);
numbers.put("Fourth", 4);
System.out.println("TreeMap: " + numbers);
System.out.println("\n使用 subMap() 方法:");
// tailMap() 使用默認(rèn)布爾值
System.out.println("使用默認(rèn)布爾值: " + numbers.subMap("Fourth", "Third"));
// tailMap() 指定布爾值
System.out.println("指定布爾值: " + numbers.subMap("Fourth", false, "Third", true));
}
}輸出結(jié)果
TreeMap: {First=1, Fourth=2, Second=2, Third=3}
使用 subMap() 方法:
使用默認(rèn)布爾值: {Fourth=4, Second=2}
指定布爾值: {Second=2, Third=3}| 方法 | 描述 |
|---|---|
| clone() | 創(chuàng)建TreeMap副本 |
| containsKey() | 搜索TreeMap指定的鍵,并返回布爾結(jié)果 |
| containsValue() | 在TreeMap中搜索指定的值并返回布爾結(jié)果 |
| size() | 返回的大小 TreeMap |
| clear() | 從中刪除所有條目 TreeMap |
在以上所有示例中,treemap元素都自然排序(以升序排列)。但是,我們也可以自定義鍵的順序。
為此,我們需要基于對樹圖中的鍵進(jìn)行排序的方式來創(chuàng)建自己的比較器類。例如,
import java.util.TreeMap;
import java.util.Comparator;
class Main {
public static void main(String[] args) {
//使用自定義比較器創(chuàng)建treemap
TreeMap<String, Integer> numbers = new TreeMap<>(new CustomComparator());
numbers.put("First", 1);
numbers.put("Second", 2);
numbers.put("Third", 3);
numbers.put("Fourth", 4);
System.out.println("TreeMap: " + numbers);
}
//創(chuàng)建一個比較器類
public static class CustomComparator implements Comparator<String> {
@Override
public int compare(String number1, String number2) {
int value = number1.compareTo(number2);
//元素以相反的順序排序
if (value > 0) {
return -1;
}
else if (value < 0) {
return 1;
}
else {
return 0;
}
}
}
}輸出結(jié)果
TreeMap: {Third=3, Second=2, Fourth=4, First=1}在上面的示例中,我們創(chuàng)建了一個treemap,將CustomComparator類作為參數(shù)傳遞。
CustomComparator類實現(xiàn)了Comparator接口。
然后重寫compare()方法,以相反順序排列元素。